【Python】ヘッダー行の追加

Question

csvから"header=None"で読み込んだ後で、最上行をヘッダーにさせることはできるのかご教示いただきたく。
ヘッダーなしで処理をした後、ヘッダーに変更して集計したいものがあります。

csvへの書き出し時にheaderを設定するのではなく、読み込んで処理をしている途中でヘッダーに変更できる術がございましたら教えてください。
お願い致します。

# 以下のデータがsample.csvに入っています。
data = """
車庫,系統,駅名,乗車人員
東京,AAA,東京,300
東京,AAA,新横浜,100
大阪,BBB,大阪,200
大阪,BBB,名古屋,100
大阪,BBB,浜松,80
"""

import pandas as pd
import numpy as np

# ヘッダーなしで読み込み
df = read_csv('sample.csv', encoding='shift-jis', header=None)
　　# 処理

# 「車庫,系統,駅名,乗車人員」をヘッダーに設定
　　# 処理

Answer 1

df.columns = df.iloc[0]
df = df.drop(df.index[0])
df['乗車人員'] = df['乗車人員'].astype(int)

でどうでしょうか。

Comments

[jiaetakoga]

ありがとうございます！
変更できました！

ちなみに、なぜ最後の行でint型に変更をしているのでしょうか、、？

[magichan]

本来、乗車人数列は数値データですので、数値型であるべきです。
にもかかわらず、ヘッダ行をデータに含めたことにより、列に文字データが含まれてしまうため、数値型ではなくObject型で扱われます。（←あまり望ましくない状態）
ですので、データから本来ヘッダとして扱われる行を削除した段階で、「乗車人数」列を数値型のデータに戻す必要があります。