合并两个数据框并选择特定列

Question

我想用列号合并两个数据帧。另外，我想将合并的数据框保存在CSV文件中，但只保存预定义的列。

column_names_1 = [..,..,...]
column_names_2 = [..,..,...]

columns = [[column_names_1,column_names_2]]
data = data_1.merge(data_2, on='Number')
data[columns].to_csv("data.csv")

此代码给出了一个错误：

TypeError: unhashable type: 'list'

Answer 1

您应该创建一个简单的列表来访问data，而不是列表列表。你需要的是 -

columns = column_names_1 + column_names_2
data = data_1.merge(data_2, on='Number')
data[columns].to_csv("data.csv")