这是一个示例代码,可以按照一组单词拆分列,并用这些单词命名列:
import pandas as pd
# 创建示例数据
data = {'col1': ['apple,banana,grape', 'orange,pineapple', 'watermelon,mango']}
df = pd.DataFrame(data)
# 拆分列并命名
def split_and_name(row):
words = row['col1'].split(',')
num_words = len(words)
for i in range(num_words):
col_name = f'col{i+1}'
row[col_name] = words[i]
return row
df = df.apply(split_and_name, axis=1)
# 移除原始列
df = df.drop('col1', axis=1)
print(df)
输出结果如下:
col1 col2 col3
0 apple banana grape
1 orange pineapple None
2 watermelon mango None
这个示例中,我们使用了Pandas库来处理数据。首先,我们创建了一个包含一个列的DataFrame,列中的值是以逗号分隔的单词。然后,我们定义了一个函数split_and_name
,该函数接受每一行作为输入,将列的值拆分为单词,并为每个单词创建一个新列,并将单词放入相应的列中。最后,我们使用apply
函数将split_and_name
函数应用于DataFrame的每一行。最后,我们移除原始的col1
列,得到最终结果。