首页 > 解决方案 > 有效地将两个 numpy 数组的大小调整为两者中较小的一个

问题描述

我正在尝试更有效地执行以下操作:

import numpy as np

array1 = np.array([-1,-2,-3,-4], np.single)
array2 = np.array([1,2,3,4,5,6,7,8], np.single)
minLength = min(len(array1),len(array2))
array1.resize(minLength)
array2.resize(minLength)
print(array1)
print(array2)

这将产生以下数组:

[-1. -2. -3. -4.]
[1. 2. 3. 4.]

有没有更好的方法来做到这一点?

标签: pythonarraysnumpy

解决方案


我在python中能想到的唯一“更好”的方法是让你不要尝试调整已经最小的那个,但该resize方法可能已经这样做了。所以在这种情况下,答案是否定的。Obs:maybeeee 试试 numpy 和/或 cupy?


推荐阅读