Python——從浮點數numpy陣列中篩選出整數


在資料清理活動中,我們有時需要取出列表中存在的整數。在本文中,我們將有一個包含浮點數和整數的陣列。我們將從陣列中移除整數,並打印出浮點數。

使用astype

astype函式將用於查詢陣列中的元素是否是整數。據此,我們將決定保留還是移除陣列中的元素,並將其儲存在結果集中。

示例

 線上演示

import numpy as np

# initialising array
A_array = np.array([3.2, 5.5, 2.0, 4.1,5])

print("Given array :\n ", A_array)

# Only integers
res = A_array[A_array != A_array.astype(int)]

# result
print("Array without integers:\n", res)

輸出

執行上面的程式碼,得到以下結果:

Given array :
   [3.2 5.5 2. 4.1 5. ]
Array without integers:
   [3.2 5.5 4.1]

使用equal和mod

在這種方法中,我們將對陣列的每個元素應用mod函式,並檢查除以結果是否為零。如果結果不為零,則認為是浮點數並保留為結果。

示例

 線上演示

import numpy as np

# initialising array
A_array = np.array([3.2, 5.5, 2.0, 4.1,5])

print("Given array :\n ", A_array)

# Only integers
res = A_array[~np.equal(np.mod(A_array, 1), 0)]

# result
print("Array without integers:\n", res)

輸出

執行上面的程式碼,得到以下結果:

Given array :
   [3.2 5.5 2. 4.1 5. ]
Array without integers:
   [3.2 5.5 4.1]

更新日期:2020年7月22日

432次瀏覽

職業生涯出色開局

完成課程即可獲得認證

開始
廣告
© . All rights reserved.