PyGame_tk 发表于 2023-5-9 19:32:04

Numpy求助

Numpy如何按索引切片

isdkz 发表于 2023-5-9 19:32:36

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

import numpy as np

arr = np.array()
indices =

sliced_arr = arr

print(sliced_arr)# 输出


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

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

import numpy as np

arr = np.array()

sliced_arr = arr

print(sliced_arr)# 输出


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

PyGame_tk 发表于 2023-5-9 19:33:11

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




谢谢,那二维数组呢

isdkz 发表于 2023-5-9 19:45:29

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

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

import numpy as np

arr_2d = np.array([,
                   ,
                   ])

# 按行切片
rows =
sliced_arr_by_rows = arr_2d

print(sliced_arr_by_rows)
# 输出:
# [
#]

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

print(sliced_arr_by_cols)
# 输出:
# [
#
#]

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

print(sliced_arr_by_rows_and_cols)
# 输出:
# [
#]


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

最后,我们演示了如何同时按行和列切片。我们首先按行切片,然后在结果上按列切片。这将生成一个新的二维数组,其中包含我们选择的行和列。
页: [1]
查看完整版本: Numpy求助