[已解决]Numpy求助

PyGame_tk

Numpy如何按索引切片

最佳答案

月排行榜 / 总排行榜

isdkz

2023-5-9 19:45:29

PyGame_tk 发表于 2023-5-9 19:33
谢谢，那二维数组呢

对于二维数组，可以分别沿着行和列方向进行切片。以下是一个例子：

import numpy as np

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

# 按行切片
rows = [0, 2]
sliced_arr_by_rows = arr_2d[rows, :]

print(sliced_arr_by_rows)
# 输出：
# [[1 2 3]
#  [7 8 9]]

# 按列切片
cols = [1, 2]
sliced_arr_by_cols = arr_2d[:, cols]

print(sliced_arr_by_cols)
# 输出：
# [[2 3]
#  [5 6]
#  [8 9]]

# 同时按行和列切片
sliced_arr_by_rows_and_cols = arr_2d[rows, :][:, cols]

print(sliced_arr_by_rows_and_cols)
# 输出：
# [[2 3]
#  [8 9]]

在这个例子中，我们首先创建了一个3x3的二维数组。然后，我们分别按行和列进行切片。对于按行切片，我们将要切片的行索引列表作为第一个索引，同时在列方向上使用冒号表示选择所有列。对于按列切片，我们将要切片的列索引列表作为第二个索引，同时在行方向上使用冒号表示选择所有行。

最后，我们演示了如何同时按行和列切片。我们首先按行切片，然后在结果上按列切片。这将生成一个新的二维数组，其中包含我们选择的行和列。

跳转到最佳答案楼层

isdkz

在NumPy中，可以使用索引和切片来访问数组的元素。要按索引切片，可以使用以下语法：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
indices = [0, 2, 4]

sliced_arr = arr[indices]

print(sliced_arr)  # 输出 [1 3 5]

在上面的例子中，我们首先创建了一个包含5个整数的NumPy数组。然后，我们定义了一个索引列表，其中包含要切片的元素的索引。最后，我们使用这个索引列表来切片原始数组，并将结果存储在一个新的数组中。

如果要按范围切片，可以使用类似于Python列表的语法：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])

sliced_arr = arr[1:4]

print(sliced_arr)  # 输出 [2 3 4]

在这个例子中，我们使用切片语法  1:4  来切片原始数组，并将结果存储在一个新的数组中。

PyGame_tk

isdkz 发表于 2023-5-9 19:32
在NumPy中，可以使用索引和切片来访问数组的元素。要按索引切片，可以使用以下语法：

谢谢，那二维数组呢

isdkz

PyGame_tk 发表于 2023-5-9 19:33
谢谢，那二维数组呢

对于二维数组，可以分别沿着行和列方向进行切片。以下是一个例子：

import numpy as np

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

# 按行切片
rows = [0, 2]
sliced_arr_by_rows = arr_2d[rows, :]

print(sliced_arr_by_rows)
# 输出：
# [[1 2 3]
#  [7 8 9]]

# 按列切片
cols = [1, 2]
sliced_arr_by_cols = arr_2d[:, cols]

print(sliced_arr_by_cols)
# 输出：
# [[2 3]
#  [5 6]
#  [8 9]]

# 同时按行和列切片
sliced_arr_by_rows_and_cols = arr_2d[rows, :][:, cols]

print(sliced_arr_by_rows_and_cols)
# 输出：
# [[2 3]
#  [8 9]]

在这个例子中，我们首先创建了一个3x3的二维数组。然后，我们分别按行和列进行切片。对于按行切片，我们将要切片的行索引列表作为第一个索引，同时在列方向上使用冒号表示选择所有列。对于按列切片，我们将要切片的列索引列表作为第二个索引，同时在行方向上使用冒号表示选择所有行。

最后，我们演示了如何同时按行和列切片。我们首先按行切片，然后在结果上按列切片。这将生成一个新的二维数组，其中包含我们选择的行和列。