python之numpy库学习常用函数(矩阵索引数据)_python numpy 获取矩阵的索引值-优快云博客

本文介绍如何使用Python的NumPy库进行高效的一维及二维数组数据索引与切片操作，包括基本的切片方法、通过布尔值索引数据、以及如何更新数组中的值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np
# 索引数据
# 一维向量
n50 = np.arange(50)
print(n50)
print('一维向量取一个数:',n50[1])
print('一维向量取一组数据(包头不包尾):',n50[1:9])
print('一维向量取这个索引之后所有的数据:',n50[1:])
print('一维向量取这个索引之前所有的数据:',n50[:7])
print('一维向量从尾部按索引取数:',n50[-3:-1])
# 二维矩阵
n510 = n50.reshape(5,10)
print(n510)
print('二维矩阵取一行:',n510[1])
print('二维矩阵取一列:',n510[:,1])
print('二维矩阵取行范围数据(包头不包尾):\n',n510[1:3])
print('二维矩阵取列范围数据(包头不包尾):\n',n510[:,1:3])
print('二维矩阵取具体某个值:',n510[1,1])
print('二维矩阵取行列范围数据(包头不包尾):\n',n510[0:2,1:4])

# 通过bool类型来取数据
mod2 = n510%2 == 0
print(mod2)
print('取出被2整除的所有数据:',n510[mod2])
print('取出第3列中被4整除的所有行数据',n510[n510[:,2] %4 == 0])
mod23 = (n510%2 == 0) & (n510%3 == 0)
print('取出被2和3都能整除的所有数据:',n510[mod23])
mod2or3 = (n510%2 == 0) | (n510%3 == 0)
print('取出被2或者3能整除的所有数据:',n510[mod2or3])

# 根据索引或者bool判断更新矩阵内值
n510[:,1] = n510[:,1]*10
print('第二列的数据都乘以10:',n510[:,1])
n510[mod2] = n510[mod2]*10
print('被2整除的数据都乘以10:', n510[mod2])