python - ファイル内の関連する列のみを選択する - python

Question

次のヘッダーとデータを含む「test.csv」ファイルがあるとします。

h1  c1  h2  h3  c2
1   0   2   3   1
3   0   2   1   0
0   1   2   3   3

関心のある列のみを選択して保存し、他のすべてを破棄するPythonの最良のオプションは何ですか?

h 列を保存することにのみ関心があると仮定すると、次の行に沿って何かを考えました。

f = open('test.csv')
s = save('new_test.csv', data = f, saveColumns=['h1','h2','h3'])´

n = load('new_test.csv')
print n

h1  h2  h3
1   2   3
3   2   1
0   2   3

Answer 1

これを行う非常に簡単な方法を見つけました：

import pandas as pd
selectColumns = ['h1','h2','h3']
table = pd.read_csv('test.csv')
tableNew = table[selectColumns]
pd.to_csv('tableNew')

Answer 2

>>> d=csv.DictReader(open("some.csv"),delimiter="\t")
>>> fields = ["h1","h2","h3"]
>>> new_rows = [[row[f] for f in fields] for row in d]
>>> d=csv.DictWriter(open("new_csv.csv","w"),fields,delimiter="\t")
>>> d.writerows(new_rows)