縦もちのデータを横にしたい

Question

やりたい事

縦にデータが入ったcsvファイルを横向きにしたいです。

csvファイル

2020	2019	2018
10	100	1000
20	200	2000
30	300	3000
40	400	4000
50	500	5000

希望する形

date	value
2020	10
2020	20
2020	30
2020	40
2020	50
2019	100
2019	200
2019	300
2019	400
2019	500
2018	1000
2018	2000
2018	3000
2018	4000
2018	5000

試してみたこと

with open('./file.csv', 'r') as f:
    reader = csv.reader(f)
    header = next(reader)
    h=[]
    t={}
    for i,hed in enumerate(header):
        #print(hed)
        t[hed]=[]
        for ii,row in enumerate(reader):
            print(row[1])
            h.append(row[ii])

全くうまくいきませんでした、、
どのようにしたらできるでしょうか？

すみませんでした。皆様のアドバイスで希望通りになりましたので下記の方法でできたので備忘録として記載します。

ppaulさんのアドバイス

with open('input.csv') as f:
    reader = csv.reader(f)
    table = list(reader)
#print(table)
header, *body = table
#print(*body)
#print(header)

result = []
for i in range(len(header)):
    for b in body:
        result.append([header[i], b[i]])

#print(result)
with open('output.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerows(result)

KojiDoiさんのアドバイス

df=pd.read_csv('./input.csv',header=0, dtype='object' )
df=df.T.stack().reset_index().drop('level_1',axis=1)

Comments

[mather]

「うまくいきませんでした」をきちんと説明しましょう。
自分で作ったコードはどのような実行結果になりましたか？何がわかりませんか？

[mn.py]

おっしゃる通りですね^^;すみませんでした。次回から気をつけます！ありがとうございました。

Answer 1

結果が欲しいだけなら、KojiDoiさんが書かれているようにpandasを使うのが楽でしょう。

こういう処理を書いてみたいなら、読み込み処理とデータの再構成と書き込み処理を分けましょう。混ぜて書くと、訳が分からなくなります。

以下のような感じです。

まず読み込み処理です。

import csv

with open('input.csv') as f:
    reader = csv.reader(f)
    table = list(reader)

print(table)

次は変換処理です。

header, *body = table
print(header)
print(body)

result = []
for i in range(len(header)):
    for b in body:
        result.append([header[i], b[i]])

print(result)

最後に書き込み処理です。

with open('output.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerows(result)

あとは確認処理です。

with open('output.csv') as f:
    print(f.read())

こういうように分けてやれば、問題点がわかりやすくなり、デバッグがはやくできるようになります。

Comments

[mn.py]

丁寧に教えていただいて、ありがとうございました！
[*]を変数の前につけると、一番最初の要素をheaderに代入、それ以降をbodyに代入となるのですね！
とても勉強になりました。
ちゃんとできました。
ありがとうございました。

Answer 2

こういうのを苦労してゼロから組んでみてもいいことはないです。ライブラリを使いましょう。
縦持ち 横持ち pythonとかでググれば有用なライブラリの情報はすぐに出てきます。
Python: PandasのDataFrameを横持ち・縦持ちに変換する

Comments

[mn.py]

ありがとうございました。
実は教えていただいたページも調べてやってみていたのですが、希望のデータにしたらマルチインデックスになってしまい、pandasに慣れておらず上手くカラム名などを扱えなく混乱してしまい、一旦使用を諦めてしまってました。なんとかpandasでもできました。おかげさまで少しpandasにも慣れました。ありがとうございました。