切片表示嵌套列表的 NumPy 数组

Question

我熟悉切片，我只是无法理解这一点，并且我尝试更改一些值来尝试说明正在发生的事情，但这对我来说没有意义。

这是示例：

import numpy
l = numpy.array([[0, 0, 0], [0, 1, 0], [1, 0, 0], [1, 1, 1]])
print(l[:,0:2].tolist())

结果：

[[0, 0], [0, 1], [1, 0], [1, 1]]

我试图将其翻译为“从索引 0 到 0,2 的切片，递增 2”这对我来说毫无意义。

Answer 1

您正在做的是多轴切片。因为 l 是一个二维数组，并且您希望对第二个维度进行切片，所以您使用逗号来指示下一个维度。

, 0:2 选择第二个维度的前两个元素。

这里有一个非常好的解释。我记得当我第一次了解它时，它很好地澄清了事情。

Answer 2

以下应该适用于普通列表。假设它是一个列表的列表，
并且所有子列表的长度都相同，那么你可以这样做（python 2）

A = [[1, 2], [3, 4], [5, 6]]
print (f"A = {A}")

flatA = sum(A, [])     # Flattens the 2D list
print (f"flatA = {flatA}")
len0 = len(A[0])
lenall = len(flatA)
B = [flatA[i:lenall:len0] for i in range(len0)] 
print (f"B = {B}")

输出将是：

A = [[1, 2], [3, 4], [5, 6]]
flatA = [1, 2, 3, 4, 5, 6]
B = [[1, 3, 5], [2, 4, 6]]