numpy genfromtxt関数で読み込んだCSVファイルを、圧縮したnpzファイルに保存する場合は、CSVのヘッダーを指定して保存する場合は以下のようにする。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
| import numpy as np
def main():
csv = 'csv_file_name'
npz = 'npz_file_name'
data = np.genfromtxt(csv, dtype=None, names=True, delimiter=',', autostrip=True, encoding='utf_8_sig')
name_list = list(data.dtype.names)
value_list = [data[name] for name in name_list]
np.savez_compressed(npz, **{name:value for name,value in zip(name_list,value_list)})
if __name__ == '__main__':
main()
|