📜  python - 包含字符串长度的列 - Python (1)

📅  最后修改于: 2023-12-03 15:04:02.923000             🧑  作者: Mango

Python - 包含字符串长度的列

在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中添加包含字符串长度的列

如果我们有一个 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中添加包含字符串长度的列的方法。