这个操作在numpy数组上的操作感觉有点麻烦,但是也没办法。
例如
1
2
3
|
a = [[1,2,3], [4,5,6], [7,8,9]] |
取 a 的 2 3 行, 1 2 列
1
2
|
c=[1,2] d =[0,1] |
若写为
1
2
3
|
b = a[c,d] output: [4 8] |
取的是 第二行第一列 和第三行第二列的数据
这并不是我们想要的结果。
正确做法是:
1
2
3
4
5
6
|
b = a[c]先取想要的行数据 b = b[:,d] print(b) output: [[4 5] [7 8]] |
这才是我们想要的结果。必须要经过这两步才能完成。
以上这篇取numpy数组的某几行某几列方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/u010882121/article/details/76045303