numpyのto.csv

Question

前提・実現したいこと

sygniteのコンペに参加しています。
以下のpredの1列目の要素を取り出しています。
その後sampに代入してcsv出力しようとするとエラーが出ます
配列の扱い方が違いますでしょうか。切り出した1列目がそのままcsvに出力させたいです。

発生している問題・エラーメッセージ

---

ValueError Traceback (most recent call last)
<ipython-input-49-31a810a64a60> in <module>
----> 1 samp[1] =preda
2 samp.to_csv("banka.csv",index=None,header=None)

ValueError: setting an array element with a sequence.

該当のソースコード

#python


pred
array([[0.41565704, 0.58434296],
       [0.41565704, 0.58434296],
       [0.94135287, 0.05864713],
       ...,
       [0.94135287, 0.05864713],
       [0.94135287, 0.05864713],
       [0.94135287, 0.05864713]])

preda = pred[:,1]
preda

array([0.58434296, 0.58434296, 0.05864713, ..., 0.05864713, 0.05864713,
       0.05864713])

/*ここでエラー*/
samp[1] =preda
samp.to_csv("banka.csv",index=None,header=None)

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[tiitoi]

samp[1] の要素数が preba の要素数と一致していないと代入できないですが、一致していることは確認しましたか？
print(samp[1].shape, preba.shape) # 一致していないと代入不可

[inroad0714]

すいませんそこもいまいちわかっていません。
samp=[]　　　←を追記
samp[1] =preda
samp.to_csv("banka.csv",index=None,header=None)

うえのように追記したところ以下のエラーになりました。
IndexError Traceback (most recent call last)
<ipython-input-59-263efb64e6b3> in <module>
1 samp=[]
----> 2 samp[1] =preda
3 samp.to_csv("banka.csv",index=None,header=None)
IndexError: list assignment index out of range

Answer 1

こんな感じですかね。

import csv

# この前にpredに質問のコードのように値の二次元配列が入れられているとする。

# csvファイルに書き込む値の一次元配列をpredから作る
preda = pred[:,1]

with open('banka.csv', 'w') as f: # 出力先のファイルを開く
    writer = csv.writer(f) # csvファイルへのwriterを作る
    writer.writerow(preda) # csvファイルへ書き込む（データが一次元配列なので、1行だけのファイルになる）

Comments

[inroad0714]

ありがとうございます。出力はできました。
出力結果が横持ちの状態になっているのですが、縦持ちにするにはどのステップを変更すればよろしいでしょうか。。
preda = pred[:,1]
この時点で既に横持ちです。