2D float numpy配列を2D int numpy配列に変換する方法は?


302

実際のnumpy配列をint numpy配列に変換するにはどうすればよいですか?配列に直接マップを使用してみましたが、機能しませんでした。

回答:


405

astypeメソッドを使用します。

>>> x = np.array([[1.0, 2.3], [1.3, 2.9]])
>>> x
array([[ 1. ,  2.3],
       [ 1.3,  2.9]])
>>> x.astype(int)
array([[1, 2],
       [1, 2]])

30
それらが驚くべき結果を持っているので、あなたが持っていないnp.infnp.nan、あなたの配列にないことを確認してください。たとえば、をnp.array([np.inf]).astype(int)出力しますarray([-9223372036854775808])
Garrett

私のマシンでnp.array([np.inf]).astype(int)np.array([-np.inf]).astype(int)、、、およびnp.array([np.nan]).astype(int)すべてが同じものを返します。どうして?
BallpointBen 2018年

1
@BallpointBen:naninfは浮動小数点値であり、意味のあるintに変換できません。あなたの前のコメントとして、驚くべき振る舞いがあり、正確な振る舞いは明確に定義されていないと思います。特定の値にマップnanしたい場合infは、自分で行う必要があります。
ブレンバーン

x.astype(int)[0] [0]はtypeではないことに注意してくださいint。ですnumpy.int32
クリスアンダーソン

これは配列をintに変換しますが、@ fhtuftの答えはそれほど予期しない結果になる可能性があることに注意してください
Nathan Musoke

66

丸めを制御するためのいくつかの派手な関数:rintfloortruncceil。フロートの丸め方、上向き、下向き、または最も近い整数への丸め方によって異なります。

>>> x = np.array([[1.0,2.3],[1.3,2.9]])
>>> x
array([[ 1. ,  2.3],
       [ 1.3,  2.9]])
>>> y = np.trunc(x)
>>> y
array([[ 1.,  2.],
       [ 1.,  2.]])
>>> z = np.ceil(x)
>>> z
array([[ 1.,  3.],
       [ 2.,  3.]])
>>> t = np.floor(x)
>>> t
array([[ 1.,  2.],
       [ 1.,  2.]])
>>> a = np.rint(x)
>>> a
array([[ 1.,  2.],
       [ 1.,  3.]])

これの1つをintにするか、他のタイプのいずれかをnumpy、astype(BrenBernで回答)にするには:

a.astype(int)
array([[1, 2],
       [1, 3]])

>>> y.astype(int)
array([[1, 2],
       [1, 2]])

2
まさに私が探していたもの。astype汎用的すぎることがよくありますが、intx-inty変換を行うときにおそらくより便利です。floatを実行したい場合-int変換で丸めの種類を選択できるのは素晴らしい機能です。
Bakuriu 2012

11
したがって、ほぼintのような7.99999整数をのようなintに安全に変換する最も簡単な方法8np.rint(arr).astype(int)
内部石

numpyでそれをuint8にする方法はありますか?
ライアン

2
@ライアンastype(np.uint8)
クリスアンダーソン

14

あなたが使うことができますnp.int_

>>> x = np.array([[1.0, 2.3], [1.3, 2.9]])
>>> x
array([[ 1. ,  2.3],
       [ 1.3,  2.9]])
>>> np.int_(x)
array([[1, 2],
       [1, 2]])

12

入力がNumpy配列になるかどうかわからない場合はasarraydtype=int代わりにwith を使用できますastype

>>> np.asarray([1,2,3,4], dtype=int)
array([1, 2, 3, 4])

入力配列にすでに正しいdtypeがある場合asarrayは、配列のコピーを回避しastypeます(指定しない限りcopy=False)。

>>> a = np.array([1,2,3,4])
>>> a is np.asarray(a)  # no copy :)
True
>>> a is a.astype(int)  # copy :(
False
>>> a is a.astype(int, copy=False)  # no copy :)
True
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.