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

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

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

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