在numpy中,可以使用numpy的函数来实现在两个数组上丢弃零值的操作。具体来说,可以使用numpy的函数numpy.nonzero()
来获取数组中非零元素的索引,然后根据这些索引来获取非零元素。
以下是一个示例代码:
import numpy as np
# 创建两个numpy数组
arr1 = np.array([1, 0, 2, 0, 3])
arr2 = np.array([0, 4, 0, 5, 0])
# 获取非零元素的索引
nonzero_idx1 = np.nonzero(arr1)
nonzero_idx2 = np.nonzero(arr2)
# 根据索引获取非零元素
nonzero_arr1 = arr1[nonzero_idx1]
nonzero_arr2 = arr2[nonzero_idx2]
# 打印结果
print("arr1中非零元素:", nonzero_arr1)
print("arr2中非零元素:", nonzero_arr2)
输出结果为:
arr1中非零元素: [1 2 3]
arr2中非零元素: [4 5]
在这个示例中,我们使用了numpy.nonzero()
函数来获取数组中非零元素的索引,然后根据这些索引来获取非零元素。最后,我们打印了两个数组中的非零元素。
对于numpy数组上的其他操作,可以参考numpy的官方文档:https://numpy.org/doc/
领取专属 10元无门槛券
手把手带您无忧上云