优化代码df1=df[df.name=='Ben'] df2=df[df.name=='Elizabeth'] df3=df[df.name=='John'] df4=df[df.name=='Smith'] df5=df[df.name=='Sherly']
时间: 2023-08-06 15:11:53 浏览: 151
代码优化方式
可以使用 Pandas 中的 `groupby` 函数来实现这个操作,代码如下:
```
grouped = df.groupby('name')
df1 = grouped.get_group('Ben')
df2 = grouped.get_group('Elizabeth')
df3 = grouped.get_group('John')
df4 = grouped.get_group('Smith')
df5 = grouped.get_group('Sherly')
```
这样就可以将原来的 5 行代码简化为 3 行。同时,如果需要处理更多不同姓名的数据,也可以很方便地加入到 `grouped` 中。
阅读全文