python - 合并两列并使用熊猫库创建新列

Question

df = pd.read_csv("school_data.csv")

    col1    col2 
0  [1,2,3]  [4,5,6] 
1  [0,5,3]  [6,2,5]

想要o/p

    col1    col2      col3
0  [1,2,3]  [4,5,6]   [1,2,3,4,5,6]
1  [0,5,3]  [6,2,5]   [0,5,3,6,2,5]

col1 和 col2 值是唯一的，使用 pandas

标签： pythonpandas

Answer 1

最简单的方法是这样做：

df['col3'] = df['col1'] + df['col2']

例子：

import pandas as pd


row1 = [[1,2,3], [4,5,6]]
row2 = [[0,5,3], [6,2,5]]

df = pd.DataFrame(data=[row1, row2], columns=['col1', 'col2'])
df['col3'] = df['col1'] + df['col2']
print(df)

输出：

        col1       col2                col3
0  [1, 2, 3]  [4, 5, 6]  [1, 2, 3, 4, 5, 6]
1  [0, 5, 3]  [6, 2, 5]  [0, 5, 3, 6, 2, 5]