用python重写csv文件

Question

我有这种结构的 csv 文件：

code1     code2     code3      name1     name2    sometnig1   something2

14355     12345     54133      part1     part12   aaaaaaaa    bbbbbbb
54782     57815     52781      part2     part22   ccccccc     ffffffff
14515     52495     52852      part3     part33   ddddddd     sssssss

我需要解析这个 csv 文件并使用我自己的标题和我需要的列创建新的 csv 文件，例如：

code_1    code_2    name_1    name_2   something_2

14355     12345     part1     part12   bbbbbbb
54782     57815     part2     part22   ffffffff
14515     52495     part3     part33   sssssss

我知道，我可以选择我需要的一列并使用 pandas 将它写入另一个文件：

df = pd.read_csv(file)
df1 = df[code_1]

但是如何选择多列并写入一个文件呢？

Answer 1

最简单的方法是只阅读您关心的列，并节省一些内存：

df = pd.read_csv(file, usecols=["code_1", "code_2", "name_1", "name_2", "something_2"])
df.to_csv("other_file.csv", index=False)