该题目来源于力扣:
1667. 修复表中的名字 - 力扣(LeetCode)
题目要求:
将姓名列的首字母大写,按照id列进行排序:
输入: Users table: +---------+-------+ | user_id | name | +---------+-------+ | 1 | aLice | | 2 | bOB | +---------+-------+ 输出: +---------+-------+ | user_id | name | +---------+-------+ | 1 | Alice | | 2 | Bob | +---------+-------+
代码实现:
import pandas as pddef 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遍历函数组合式使用