该题目来源于力扣:
题目要求:
将姓名列的首字母大写,按照id列进行排序:
输入: Users table: +---------+-------+ | user_id | name | +---------+-------+ | 1 | aLice | | 2 | bOB | +---------+-------+ 输出: +---------+-------+ | user_id | name | +---------+-------+ | 1 | Alice | | 2 | Bob | +---------+-------+
代码实现:
import pandas as pd
def fix_names(users: pd.DataFrame) -> pd.DataFrame:
users['name']=users['name'].str.capitalize()
users=users.sort_values('user_id')
return users
拓展:
功能强大的pandas库可以对字符串类型的数据做多样的操作,接下来我做一个拓展补充
- 大小写转换
.str.lower()
:转换为小写.str.upper()
:转换为大写.str.title()
:每个单词的首字母大写
- 去空格
.str.strip()
:去除两侧的空格.str.lstrip()
:去除左侧的空格.str.rstrip()
:去除右侧的空格
- 分割与连接
.str.split pat
:按指定模式分割字符串.str.cat(sep='')
:连接字符串,可以用指定分隔符
- 空格检测
.str.contains(' ')
:检测字符串是否包含空格(返回布尔值)
- 替换
.str.replace(a, b)
:替换字符串中的子串.str.replace(to_replace, value, regex=True)
:使用正则表达式替换
- 计算出现次数
.str.count('pattern')
:计算字符串中某个模式出现的次数
- 重复
.str.repeat(n)
:重复字符串
- 提取子串
.str.slice(start, end)
:提取子串.str.get(loc)
:通过位置获取字符串中的某个字符
- 长度
.str.len()
:计算字符串的长度
- 判断是否包含某个字符串
.str.contains('pattern')
:判断是否包含某个字符串
- 正则表达式匹配
.str.match('pattern')
:检查是否与正则表达式匹配
# 使用 .str.match('^[a-z]p...') 检查每个字符串是否以小写字母开头,后面跟着任意两个字符和字母p,再跟着任意一个字符
match_pattern = df['column'].str.match('^[a-z]p...')
- 应用函数
.str.apply(func)
:对每个字符串应用一个函数,常常和lambda遍历函数组合式使用