for文でnp.arrayのデータ処理を行いたい

Question

np.arrayの中の変数が不適切です。

pythonを使って簡単なデータの処理を行おうとしているのですが、おそらくfor文でarrayに入るiが不適切と止められてしまいます。どうすればよいでしょうか。

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

arrays used as indices must be of integer (or boolean) type

該当のソースコード

import numpy as np
import pandas as pd

hissu = []
sonota =[]
i = 1

data = pd.read_csv("credit.csv")
data = data.loc[:,['a','b']]
data = np.array(data)

for i in data: 


    if data[i,0].isdigit() == True:
        hissu.append(data[i,1])
    else:
        sonota.append(data[i,1])
    i = i+1
return hissu, sonota

試したこと

ここに問題に対して試したことを記載してください。

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

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

Comments

[tachikoma]

credit.csvの中身の例を追記してもらえませんでしょうか。

Answer 1

pandasのみで処理するとこんな感じの記述となります。

import pandas as pd

data = pd.DataFrame({'a':['1','2','A','B','3','C'],
                     'b':['h1','h2','s1','s2','h3','s3']})

is_digit = data['a'].str.isdigit()
hissu = data.loc[is_digit, 'b'].tolist()
sonota = data.loc[~is_digit, 'b'].tolist()

print(hissu)
print(sonota)

Answer 2

data.ndimが2だという前提で。
for i in data:で巡回しているので、iはおそらく数値ではありません。
未確認ですが、次のようにすれば解決するかと。

for datum in data: 
    if datum[0].isdigit():
        hissu.append(datum[1])
    else:
        sonota.append(datum[1])

Comments

[LouiS0616]

numpyではなくpandas側で解決するべき問題に思えてきた。

Answer 3

やりたいことがうまく把握できていませんが、とりあえず読込データが以下のような感じなら～in dataではなく～in range(data.shape[0]):で動作はするはずです。

import numpy as np
import pandas as pd

hissu = []
sonota =[]

data = pd.DataFrame({'a':['1','c'], 'b':['h','s']})
data = data.loc[:,['a','b']]
print(data)

data = np.array(data)
print(data)

for i in range(data.shape[0]):
    if data[i,0].isdigit() == True:
        hissu.append(data[i,1])
    else:
        sonota.append(data[i,1])

print(hissu)
print(sonota)