Python:
CSVとの作業
方法
PythonはCSVファイルを扱うために組み込みのcsv
モジュールを提供し、これを使ってからの読み取りと書き込みが簡単になります。より堅牢で複雑なデータ操作には、サードパーティのライブラリpandas
が非常に人気です。
csv
モジュールを使用する
CSVファイルを読む
import csv
with open('sample.csv', mode='r') as file:
csv_reader = csv.reader(file)
for row in csv_reader:
print(row)
sample.csv
が以下を含むと仮定します:
name,age,city
John,22,New York
Jane,28,Los Angeles
出力:
['name', 'age', 'city']
['John', '22', 'New York']
['Jane', '28', 'Los Angeles']
CSVファイルに書き込む
import csv
rows = [['name', 'age', 'city'], ['Jack', '33', 'Chicago'], ['Emily', '41', 'Denver']]
with open('output.csv', mode='w', newline='') as file:
writer = csv.writer(file)
writer.writerows(rows)
output.csv
を以下で作成または上書きします:
name,age,city
Jack,33,Chicago
Emily,41,Denver
CSVに対してpandas
を使用する
pandas
はデータ操作を簡単にする強力なライブラリで、CSVファイルなどのさまざまなデータ形式を扱うことを簡略化します。
pandasをインストールする
pip install pandas
pandasでCSVファイルを読む
import pandas as pd
df = pd.read_csv('sample.csv')
print(df)
出力:
name age city
0 John 22 New York
1 Jane 28 Los Angeles
pandasでCSVファイルに書き込む
import pandas as pd
df = pd.DataFrame({'name': ['Jack', 'Emily'], 'age': [33, 41], 'city': ['Chicago', 'Denver']})
df.to_csv('output_pandas.csv', index=False)
output_pandas.csv
を以下で作成または上書きします:
name,age,city
Jack,33,Chicago
Emily,41,Denver