Numpy：使用索引、切片提取一维数组、多维数组内数据

一、使用索引提取值

print("---------------书写格式2------------------------")
# dtype="U75"   ：表示以字符串类型导入（75是导入的字符个数为75）
# skip_header=1 ：表示从txt的第1行导入，即表头（第0行）不导入
world_alcohol = numpy.genfromtxt("world_alcohol.txt", delimiter=",", dtype="U75", skip_header=1)

print(world_alcohol)

print("-----------使用索引提取单个值------------------")
uruguay_other_1986 = world_alcohol[1,4]   # 提取第1行第4列的值
third_country = world_alcohol[2,2]        # 提取第2行第2列的值

print (uruguay_other_1986)
print (third_country)

结果图：

提取值后结果：

二、使用切片提取一维数组、多维数组数据

print("---------------使用切片提取一维数组内的数据-----------------")
vector = numpy.array([5, 10, 15, 20])
print(vector[0:3])   # 一维数组：提取第0列到第2列所有的值，返回：[ 5 10 15]


print("----------------使用切片提取二维数组内的数据--------------------------")
matrix = numpy.array([
                    [5, 10, 15], 
                    [20, 25, 30],
                    [35, 40, 45]
                 ])
print(matrix[:,1])   # 二维数组：提取所有行的第1列的值，返回：[10 25 40]
print("---------")
print(matrix[:,0:2])   # 提取所有行的第0列导第1列的值
print("---------")
print(matrix[1:3,0:2])  # 提取第1行到第2行的第0列到第1列的值

提取后结果