pythonを使ってcsvのファイルを生成する方法

pythonを使ったcsvのファイルを生成する方法を紹介します。

この記事では、csvの読み込みは扱いません。csvファイルを作成する方法を紹介しています。

csvを扱うための便利なライブラリをインストールします。

pip install pandas

csvを読み書きするためにpandasをインストールします。

pandasをインストールするとnumpyもインストールされます。

csvファイルを生成する関数を作成します。

output_csv_root = 'csv/files'
def gen_csv(dataset, file_name):
    df = pd.DataFrame(dataset)
    file_path = os.path.join(output_csv_root, file_name + '.csv')
    df.to_csv(file_path)

gen_csvを実行すると、{output_csv_root}/{filename}.csvが作成されます。

datasetに指定する値は、二次元の配列を指定します。

[[1,2,3], [4,5,6]]

これがこうなります。

#, 1, 2, 3
1, 1, 2, 3
2, 4, 5, 6

一行目の#, 1, 2, 3は見出しです。

この見出しは、自分で指定することもできます。

指定する場合は、dataFrameを作成する際にcolumnsを指定します。

df = pd.DataFrame([rows], columns=['A', 'B', 'C'])

また、行番号が必要ない場合は、このようにindexをFalseにします。

df.to_csv(file_path, index=False)