效果图
代码
import numpy as np list = [ [1,2,3,4,5], [6,7,8,9,10], [11,12,13,14,15], [16,17,18,19,20], [21,22,23,24,25] ] array = np.array(list) print(array) 输出: [[ 1 2 3 4 5] [ 6 7 8 9 10] [11 12 13 14 15] [16 17 18 19 20] [21 22 23 24 25]]
a = array[:,2] print(array[:,2]) print(a[0]) print(type(a)) print(a.shape) 输出: [ 3 8 13 18 23] 3 <class 'numpy.ndarray'> (5,) 总结: 输出的结果是所有行的第2列,取出来的元素被组成一个一维数组
a = array[:,2:3] print(array[:,2:3]) print(a[0],a[0][0]) print(type(a)) print(a.shape) 输出: [[ 3] [ 8] [13] [18] [23]] [3] 3 <class 'numpy.ndarray'> (5, 1) 总结: 输出的结果还是所有行第2列,但是第2列的元素分别被当做一个一维数组返回, 最终返回的是一个二维数组