u++の備忘録

言語処理100本ノック 2020「12. 1列目をcol1.txtに,2列目をcol2.txtに保存」

問題文

nlp100.github.io

問題の概要

pandasでデータを読み込み、列ごとにファイルに吐き出します。

import pandas as pd


df = pd.read_csv('ch02/popular-names.txt', sep='\t', header=None)
df[0].to_csv('ch02/col1.txt', index=False, header=None)
df[1].to_csv('ch02/col2.txt', index=False, header=None)