如何根据数组的长度重新排列存储在列表中的numpy数组

Question

我有一个 numpy 数组列表，并希望根据每个数组的长度重新排列它。这是我的清单：

nums=[np.array([[2.], [3.]]), np.array([[8.], [7.], [1.]]),\
      np.array([[1.], [0.], [5.]]), np.array([[8.], [9.]]), np.array([[7.], [6.]])]

当数组的长度发生变化时，这一点很重要。长度列表是：

lns=[2, 3, 3, 2, 2]

基于lns，长度从第一个数组变为第二个数组。第一个数组更短，我想要它的所有行。我想要2（较短数组的长度）行较长（第二）行。在这种情况下，较短数组 (2.5) 的平均值小于较长数组 (5.3) 的平均值，因此我提取了较长数组的第一行 2。我想要第二个和第三个数组，因为它们的长度是最大的。从第三到第四长度变化。同样，我想要所有更短的行和2（更短的长度，即第四行）。但是，较短 (8.5) 的平均值高于较长 (2.)。所以，我需要较长的最后一行2。最后，我想重新排列我的列表以获得以下列表：

final=[np.array([[2.], [3.], [8.], [7.]]),\ # all rows of first and first 2 rows of second
       np.array([[8.], [7.], [1.], [1.], [0.], [5.]]),\
       np.array([[0.], [5.], [8.], [9.], [7.], [6.]])]# last 2 rows of third and all rows of next short arrays

感谢任何帮助我在 python 中做我想做的事。

Answer 1

1。首先，我使用pairwise() recipe from itertools 一次获取两个连续元素，而不是使用索引：

import itertools

# https://docs.python.org/3/library/itertools.html#itertools-recipes
def pairwise(iterable):
    "s -> (s0,s1), (s1,s2), (s2, s3), ..."
    a, b = itertools.tee(iterable)
    next(b, None)
    return zip(a, b)

2。我创建了一个辅助函数first_or_last()，它返回一个slice object，它可以是used on a numpy array，并应用于两个相邻数组中较大的一个进行比较³。根据您上面描述的逻辑，需要应用哪个切片取决于平均值/平均值。如果需要，您可以将其简化为两行函数，我已将较长的版本用于 cmets/explanation。

def first_or_last(smaller, larger):
    """return a slice object for either the first-n items or
    the last-n items of the larger array
    """
    size = len(smaller)
    if np.mean(smaller) < np.mean(larger):
        # take first-`size` elements
        return slice(None, size)
    # take last-`size` elements
    return slice(-size, None)  # don't skip the extra None, needed for axes
    # or just
    # return slice(None, size) if np.mean(smaller) < np.mean(larger) else slice(-size, None)

3 .重新排列的代码：

nums = [np.array([[2.], [3.]]), np.array([[8.], [7.], [1.]]),
        np.array([[1.], [0.], [5.]]), np.array([[8.], [9.]]),
        np.array([[7.], [6.]])]

final = []
for ar1, ar2 in pairwise(nums):
    l1, l2 = len(ar1), len(ar2)
    if l1 == l2:  # same length
        final.append(np.append(ar1, ar2, axis=0))
        continue
    # different lengths
    arrs = (ar1, ar2[first_or_last(ar1, ar2)]) if l1 < l2 else (ar1[first_or_last(ar2, ar1)], ar2)
    final.append(np.append(*arrs, axis=0))

final 的值：

[array([[2.], [3.], [8.], [7.]]),
 array([[8.], [7.], [1.], [1.], [0.], [5.]]),
 array([[0.], [5.], [8.], [9.]]),
 array([[8.], [9.], [7.], [6.]])]

---

注意事项：

1. 此输出与您所拥有的 final 不同。请参阅item #9 in my comment，因为我看不到使结果包含 3 个项目而不是 4 个项目的规则/逻辑。

   9. 另一个逻辑不一致：对于数组 3 (len 3) 和数组 4 (len 2)，结果应该是0 5 8 9，结果中的最后一项应该是@ 987654338@。为什么最后会合并为0 5 8 9 7 6？他们应该是分开的。对于第 3、第 4 和第 5 个数组而不是第 1、第 2 和第 3 个数组合并它们的逻辑是什么？

2. 此解决方案不会为 your previous/original question 提供错误，但它似乎也不是正确的输出 - 尽管它确实遵循相同的规则。

3. NumPy 数组切片返回原始数组的视图，而不是副本。