pythonを用いて、偏りのあるラベルデータを生成させた後に、CSVに保存することに苦慮しております。

Question

pythonを用いて、偏りのあるラベルデータを生成させた後に、CSVに保存することに苦慮しております。

# coding: utf-8

import urllib2
import codecs
import numpy as np
import pandas as pd

f1 = codecs.open('panda_1.csv', 'w', 'utf-8')

# 偏りのあるラベルデータを生成する
data = pd.DataFrame(np.random.randn(10, 3))
data['label'] = [1, 1, 1, 1, 1, 0, 0, 1, 1, 1]
pd.DataFrame(data)
print pd.DataFrame(data)

cols = [pd.DataFrame(data)]
f1.write(",".join(str(v) for v in cols)+ "\n") 

上記のソースコードを動かしますと、以下のような結果になります。

      0         1         2  label

0 0.656896 -1.141412 1.675323 1
1 0.006246 -0.838113 -0.089666 1
2 1.324973 0.960406 -1.017579 1
3 -0.619690 -0.865968 0.673617 1
4 -1.408692 0.574759 0.544791 1
5 -1.139060 -0.425326 -0.678609 0
6 0.312005 -0.085453 0.857054 0
7 -1.003243 -0.309490 -0.585781 1
8 0.759179 1.027915 0.499001 1
9 -1.884306 -0.669742 -1.412429 1

御覧のように、9行のデータが、ひとつのセルに入った状態で吐き出されました。

目的は、吐き出したCSVの各データが、エクセルの各マス目に、きちんと入ってそろっていることですが、

最後のソースコードの

cols = [pd.DataFrame(data)]
f1.write(",".join(str(v) for v in cols)+ "\n") 

この箇所の改変が、うまく出来ずに苦慮しております。

先輩方の御教示、よろしくお願いいたします。

Answer 1

提示されたコードはかなり冗長な気がします。

pd.DataFrame(data) の行も
cols = [pd.DataFrame(data)]　の行も不要です

また、DataFrameからCSVへの吐き出しはDataFrame.to_csv()をお使いください。

# coding: utf-8

import numpy as np
import pandas as pd

# 偏りのあるラベルデータを生成する
data = pd.DataFrame(np.random.randn(10, 3))
data['label'] = [1, 1, 1, 1, 1, 0, 0, 1, 1, 1]
data.to_csv('panda_2.csv', index=None)

---

一応補足

DataFrameから２次元リスト変換するには

cols = [pd.DataFrame(data)]

と記述するのではなく、一般的に

cols = data.as_matrix().tolist()

と記述します。

as_matrix()でnumpy配列に変換した後に、tolist()でリスト化です

Comments

[akakage13]

magichan様、いつも懇切丁寧な御教示、ありがとうございます。
今回も、眼からうろこの御説明、本当にありがとうございました。
今後とも、よろしくお願いいたします。

Answer 2

colsの中に要素が１つしかない状況になってるからですね。

cols = list(pd.DataFrame(data))
f1.write(",".join(str(v) for v in cols)+ "\n") 

このようにしてみてください。