在numpy中,可以使用以下方法來通過字段名進行索引:
import numpy as np
# 創建一個帶有字段名的結構化數組
data = np.array([(1, 'apple', 0.5), (2, 'banana', 0.3)],
dtype=[('id', int), ('fruit', 'U10'), ('price', float)])
# 將數組轉換為np.recarray類型
rec_data = np.rec.array(data)
# 使用字段名作為屬性來索引記錄
print(rec_data.id)
print(rec_data.fruit)
print(rec_data.price)
輸出:
[1 2]
['apple' 'banana']
[0.5 0.3]
np.ndarray
的方法np.ndarray.item()
和np.ndarray.itemset()
來通過字段名進行索引和賦值。首先,使用字段名作為索引來獲取記錄,然后使用.item()方法獲取相應的值,或使用.itemset()方法設置相應的值。例如:import numpy as np
# 創建一個帶有字段名的結構化數組
data = np.array([(1, 'apple', 0.5), (2, 'banana', 0.3)],
dtype=[('id', int), ('fruit', 'U10'), ('price', float)])
# 使用字段名作為索引來獲取記錄
record = data[0]
# 通過字段名進行索引和賦值
print(record.item('id'))
record.itemset('id', 3)
print(record.item('id'))
輸出:
1
3
需要注意的是,使用這種方法索引字段名時,只能獲取或設置單個記錄的值,不能同時獲取或設置多個記錄的值。