📅  最后修改于: 2023-12-03 15:04:02.923000             🧑  作者: Mango
在Python中,我们可以使用 len()
函数获取一个字符串的长度,但是如果我们想要在一个列表或者一个DataFrame中添加一个包含字符串长度的列,该怎么做呢?
假设我们有一个包含字符串的列表,我们可以使用一个简单的列表推导式来添加一个包含字符串长度的列:
strings = ['apple', 'banana', 'orange', 'peach']
lengths = [len(s) for s in strings]
data = list(zip(strings, lengths))
print(data)
输出:
[('apple', 5), ('banana', 6), ('orange', 6), ('peach', 5)]
我们使用了 zip
函数将字符串列表和长度列表打包在一起,然后转换成列表。
如果我们有一个 DataFrame,我们同样可以使用 apply
方法来添加一个包含字符串长度的列:
import pandas as pd
df = pd.DataFrame({
'fruit': ['apple', 'banana', 'orange', 'peach']
})
df['length'] = df['fruit'].apply(len)
print(df)
输出:
fruit length
0 apple 5
1 banana 6
2 orange 6
3 peach 5
我们使用了 apply
方法来将每个字符串应用 len
函数,然后将结果赋值给新的列。
以上就是在Python中添加包含字符串长度的列的方法。