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次浏览

职业生涯出色开局

完成课程即可获得认证

开始
广告