python int型に変換できないものをアレイから除きたい

Question

Python3.9.7を使っています。

import numpy as np

list = ['1','a','b','2']
arr =np.array(list)
print(arr)

例えばこのアレイを例にして、int型に変換できない'a'と'b'を除いたアレイを作るには、どのようなコードを書けばよいでしょうか？

出力は以下のようにしたいです。

[1,2]

ご回答お待ちしております。よろしくお願いいたします。

Answer 1

以下でできます。

try文については、公式ドキュメント 8.4. try 文
と、Pythonの例外処理（try, except, else, finally）をお読みください。

>>> def to_num(s):
...     try:
...         return int(s)
...     except ValueError:
...         return s
...
>>> import numpy as np
>>>
>>> lst = ['1','a','b','2']
>>> arr = np.array([to_num(s) for s in lst if type(to_num(s)) == int])
>>> print(arr)
[1 2]

Comments

[eri555]

できました！ありがとうございます。