Pythonの配列を条件指定で抽出した値の配列型を整数型に変換するには？

Question

下記のようなソースコードがあり、
c1は文字列で返され、
c2は配列で返されます。

ある関数fにこれらcを使うとすると、
f(c1)ではエラーとならず,f(c2)ではエラーとなり、
c2を整数にCastしたいのですが、
どのようなソースコードを追加すればよいでしょうか。
c2が５と表示されればOKです。

import numpy as np

np_items=np.array([["P00",1, 2, 3], ["P01",4, 5, 6]])

print(np_items)
c1=np_items[1,2]#Stringで出る。
print(c1)
print(type(c1))

c2 =np_items[np_items[:,0]=='P01',2]
print(c2)#['5']と出る。整数にしたい。５と表示したい。
print(type(c2))

環境は、Google Colaboratoryです。

よろしくお願いします。

Answer 1

配列で返されるc2から要素を取り出すだけなら、

-c2 =np_items[np_items[:,0]=='P01',2]
+c2 =np_items[np_items[:,0]=='P01',2][0]

で済みますし、文字列が得られるのを整数にしたいのであれば、c1も含めて、文字列が得られる式をint(...)で囲めばいいのでは。

Comments

[occy]

ありがとうございました。理解しました。

c2 =np_items[np_items[:,0]=='P01',2]
print(c2[0])
という事ですね。ベストアンサーに選ばせて頂きます。

Answer 2

arrayのdtypeにobjectを指定してやれば良さそうです。

import numpy as np

np_items = np.array(
    [["P00", 1, 2, 3], ["P01", 4, 5, 6]],
    dtype=object)

Comments

[occy]

ご協力ありがとうございます。
これだと、結果が変わりませんでした。

[bsdfan]

質問をちゃんと理解していませんでした。
これは、intが文字列になってしまうことに対する回答です。
配列になるのが問題ならDaregadaさんの回答になりますね。

[occy]

ご確認ありがとうございました。
配列をよく理解していなくて、私の質問の仕方が悪かったです。