[已解决]Numpy求助

PyGame_tk

Numpy如何按索引切片

最佳答案

月排行榜 / 总排行榜

isdkz

2023-5-9 19:45:29

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

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

1. import numpy as np
   
2. 
   
3. arr_2d = np.array([[1, 2, 3],
   
4.                    [4, 5, 6],
   
5.                    [7, 8, 9]])
   
6. 
   
7. # 按行切片
   
8. rows = [0, 2]
   
9. sliced_arr_by_rows = arr_2d[rows, :]
   
10. 
    
11. print(sliced_arr_by_rows)
    
12. # 输出：
    
13. # [[1 2 3]
    
14. #  [7 8 9]]
    
15. 
    
16. # 按列切片
    
17. cols = [1, 2]
    
18. sliced_arr_by_cols = arr_2d[:, cols]
    
19. 
    
20. print(sliced_arr_by_cols)
    
21. # 输出：
    
22. # [[2 3]
    
23. #  [5 6]
    
24. #  [8 9]]
    
25. 
    
26. # 同时按行和列切片
    
27. sliced_arr_by_rows_and_cols = arr_2d[rows, :][:, cols]
    
28. 
    
29. print(sliced_arr_by_rows_and_cols)
    
30. # 输出：
    
31. # [[2 3]
    
32. #  [8 9]]
    

复制代码

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

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

跳转到最佳答案楼层

isdkz

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

1. import numpy as np
   
2. 
   
3. arr = np.array([1, 2, 3, 4, 5])
   
4. indices = [0, 2, 4]
   
5. 
   
6. sliced_arr = arr[indices]
   
7. 
   
8. print(sliced_arr)  # 输出 [1 3 5]
   

复制代码

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

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

1. import numpy as np
   
2. 
   
3. arr = np.array([1, 2, 3, 4, 5])
   
4. 
   
5. sliced_arr = arr[1:4]
   
6. 
   
7. 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
谢谢，那二维数组呢

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

1. import numpy as np
   
2. 
   
3. arr_2d = np.array([[1, 2, 3],
   
4.                    [4, 5, 6],
   
5.                    [7, 8, 9]])
   
6. 
   
7. # 按行切片
   
8. rows = [0, 2]
   
9. sliced_arr_by_rows = arr_2d[rows, :]
   
10. 
    
11. print(sliced_arr_by_rows)
    
12. # 输出：
    
13. # [[1 2 3]
    
14. #  [7 8 9]]
    
15. 
    
16. # 按列切片
    
17. cols = [1, 2]
    
18. sliced_arr_by_cols = arr_2d[:, cols]
    
19. 
    
20. print(sliced_arr_by_cols)
    
21. # 输出：
    
22. # [[2 3]
    
23. #  [5 6]
    
24. #  [8 9]]
    
25. 
    
26. # 同时按行和列切片
    
27. sliced_arr_by_rows_and_cols = arr_2d[rows, :][:, cols]
    
28. 
    
29. print(sliced_arr_by_rows_and_cols)
    
30. # 输出：
    
31. # [[2 3]
    
32. #  [8 9]]
    

复制代码

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

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