假设有两个csv文件:file1.csv和file2.csv,其中file1.csv包括2列(A和B),file2.csv包括2列(C和D)。现在我们想将列A、B、C和D按照某个顺序组合成一个新的csv文件。
我们可以通过pandas库的concat函数来将两个文件按行合并,再通过reindex函数按照想要的列顺序对列进行排序,最后再写入新的csv文件。
代码示例:
import pandas as pd
# 读取两个csv文件
df1 = pd.read_csv("file1.csv")
df2 = pd.read_csv("file2.csv")
# 合并两个文件
df = pd.concat([df1, df2], axis=1)
# 按照指定顺序重新排序列
df = df.reindex(columns=["A", "C", "D", "B"])
# 将结果写入新的csv文件
df.to_csv("new_file.csv", index=False)
在上述代码中,axis=1代表按列合并,reindex函数中的columns参数指定了排序后列的顺序,index=False表示不将索引写入新的csv文件。