在python的dataframe中添加一列,内容为前一列元素的长度
时间: 2024-05-06 15:21:44 浏览: 152
pyspark给dataframe增加新的一列的实现示例
可以使用apply函数来实现这个功能,具体的代码如下所示:
```
import pandas as pd
# 创建一个包含测试数据的DataFrame
df = pd.DataFrame({'A': ['apple', 'banana', 'orange'], 'B': [1, 2, 3]})
# 使用apply函数来计算前一列元素的长度,并添加为新的一列
df['C'] = df['A'].apply(lambda x: len(x))
print(df)
```
运行上述代码后,将会输出以下结果:
```
A B C
0 apple 1 5
1 banana 2 6
2 orange 3 6
```
可以看到,我们成功地添加了一列C,其内容为前一列A中每个元素的长度。
阅读全文