首頁 > 軟體

python numpy查詢定位賦值數值所在行列

2022-09-26 14:04:59

根據條件篩選行(篩選)

篩選矩陣中第7列值為5的行

B = A[ A[:,6] == 5]

篩選矩陣中第7列大於5的行

B = A[ A[:,6] > 5]

Numpy基礎操作

根據行列號取值(查詢)

取第2行第2列的數位

import numpy as np
#產生3行4列的矩陣
x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

y=x[1,1]
print(y)

擷取前幾行前幾列

import numpy as np
#產生3行4列的矩陣
x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

#[a:b,c:d] 表示取a-b行,c-d列。a、c可以省略,表示從0開始
#取x的前兩行(所有列)
y=x[:2] # 等價於 y=x[:2,:],等價於 y=x[0:2,...]
print(y)

#取x的前兩列(的所有行)
z=x[:,:2] # 等價於 z=x[:,0:2],等價於 z=x[...,0:2]
print(z)

擷取某幾行某幾列

import numpy as np
#產生3行4列的矩陣
x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

#[a:b,c:d] 表示取a-b行,c-d列。a、c可以省略,表示從0開始
#取x的第2行第2-3列
y=x[1,1:3]

根據值求行列號(定位)

輸出某行最大值所在的列索引

import numpy as np
#產生3行4列的矩陣
x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

#  取出x中元素最大值所對應的索引,按照a[0][1]中的a[1]方向,即行方向搜尋最大值。
y = np.argmax(x, 1)
print(y)

輸出某列最大值所在的行索引

#  取出x中元素最大值所對應的行號
y = np.argmax(x, 0)
print(y)

最大值所在行列號

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

# where返回一個長度為2的元組,第一個元素儲存的是行號,第二個元素儲存的是列號
y = np.where(x == np.max(x))
print(y)
print("最大值所在行:",y[0],"最大值所在列:",y[1])

按行/列求和(求和)

按行求和,將和新增到矩陣最後一列

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

y = np.sum(x, axis=1)
print(y)

xy = np.hstack((x, y.reshape(x.shape[0], -1)))
print(xy)

按列求和,將和新增到矩陣最後一行

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

z = np.sum(x, axis=0)
print(z)

xz = np.vstack((x, z))
print(xz)

賦值

按條件賦值

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

x[x<=5]=0 # 將小於5的數賦值為0
print(x)

where()

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

# results = np.where(condition, x, y)
# 當條件為真時,對應位置返回x中的值,條件不成立則返回y中的值
y = np.where(x>5,x,0)  #滿足大於5的值設定為x,不滿足的設為0
print(y)

邏輯運算

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

print(x>5) # x>5的為Ture,否則為False

給指定 行 / 列 / [行,列] 賦值

x=np.arange(0,12)
x=x.reshape((3,4))
print(x)

x[1,1] = 999 # 指定行列號賦值
print(x)

x[1] = 123 # 指定行賦值
print(x)

x[:,1] = 321 # 指定列賦值
print(x)

到此這篇關於python numpy查詢定位賦值數值所在行列的文章就介紹到這了,更多相關python numpy 內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com