python 辞書型の書き込みについて

Question

d = {"a":[], "b":[]}

上記のような辞書型のデータをｃｓｖに書き込み、

a b
1000 2355
2554 4544
4445 7556
4655
7755

例えば、上記のようなデータを作成したいです。

with open('/python/rads.csv', 'w') as f:
writer = csv.DictWriter(f, ['a', 'b'])
writer.writeheader()
writer.writerows([d])

としたところ、保存はできましたが、１つのセルにいくつもの数字が入ってしまい大変読みにくくなってしまいました。
上記のように、１つのセルにデータを１つ入れるためにはどのように記述すればよいのでしょうか。

加えて、ｓｅｔ関数を使って、aとbをから重複した数字を抜き出したいのですが、うまくできません。
set(d["a"]) &　set(d["b"])
という記述はどこが間違っているのでしょうか。

よろしくお願いします。

Answer 1

dictに含まれている各配列の長さが同じであるならば、Pandasを使って

import pandas as pd
import numpy as np

d = {'a' : [1000,2554,4445,4655,7755],'b':[2355,4544,7556,np.nan,np.nan]}

pd.DataFrame(d).to_csv('out.csv')

でよいのではと思います。

出力ファイル(out.csv)

,a,b
0,1000,2355.0
1,2554,4544.0
2,4445,7556.0
3,4655,
4,7755,

各配列の長さが違うのであれば、列毎にSeries化して結合することででDataFrameを作ることで、同じ方法を使うことができます。

import pandas as pd

d = {'a' : [1000,2554,4445,4655,7755],'b':[2355,4544,7556]}

df_a = pd.Series(d['a'], name='a')
df_b = pd.Series(d['b'], name='b')
pd.concat([df_a, df_b], axis=1).to_csv('out.csv')

Comments

[TheBullchannel]

pandasを使うと、aとbの長さが違うという様なエラーが出ていたので、できないものだと思っていましたが、このようなやり方もあるのですね！
勉強になりました。ありがとうございました。

Answer 2

要求するようなCSVを作成したいということであれば、dは、このような内容のリストになっている必要があります。

d = [{'a': 1000, 'b': 2355},
     {'a': 2554, 'b': 4544},
     {'a': 4445, 'b': 7556},
     {'a': 4655, 'b': ''},
     {'a': 7755, 'b': ''}]

値が無い列のbの値を空白にしましたが、bそのものが無くてもいいです。

なので、

• このような形のデータを用意する
• もとのデータをこのような形に変形する処理を作る
• 書込みのループの中でデータを生成する

などの対応をすればよいのではないかと思います。

Comments

[TheBullchannel]

参考にさせていただきます！
ありがとうございました。