我有 6 個(gè)數(shù)組，名為 , x1,......x6 我從 'npz' 文件中讀取。我需要對(duì)每個(gè)數(shù)組執(zhí)行一些數(shù)學(xué)運(yùn)算并將其存儲(chǔ)到 10 個(gè)新數(shù)組中。我正在以一種非常簡單的方式逐步進(jìn)行。要讀取文件并存儲(chǔ)變量，files = np.load("particle.npz")x1 = files['x1']x2 = files ['x2']x3 = files['x3']x4 = files ['x4']x5 = files['x5']x6 = files ['x6']從前一個(gè)數(shù)組創(chuàng)建另一個(gè)數(shù)組，pox1= x1[:,0]pox2= x2[:,0]pox3= x3[:,0]pox4= x4[:,0]pox5= x5[:,0]pox6= x6[:,0]然后創(chuàng)建一些新數(shù)組，sq_diff_x1 = np.zeros(40002) sq_diff_x2 = np.zeros(40002)sq_diff_x3 = np.zeros(40002)sq_diff_x4 = np.zeros(40002)sq_diff_x5 = np.zeros(40002)sq_diff_x6 = np.zeros(40002)最后使用 for 循環(huán)執(zhí)行計(jì)算并存儲(chǔ)到新數(shù)組中，for i in range (len(x1)-1): sq_diff_x1[i] = (pox1[i]-pox1[0])**2 sq_diff_x2[i] = (pox1[i]-pox1[0])**2 sq_diff_x3[i] = (pox1[i]-pox1[0])**2 sq_diff_x4[i] = (pox1[i]-pox1[0])**2 sq_diff_x5[i] = (pox1[i]-pox1[0])**2 sq_diff_x6[i] = (pox1[i]-pox1[0])**2代碼工作正常但是有沒有其他方法可以通過不一個(gè)一個(gè)地分配所有內(nèi)容來自動(dòng)完成它？因?yàn)槭褂梦业姆椒ê芎唵?，但是?dāng)我需要處理 100 個(gè)數(shù)組時(shí)會(huì)非常耗時(shí)。所以需要一些自動(dòng)化的東西。
2 回答
慕碼人8056858
TA貢獻(xiàn)1803條經(jīng)驗(yàn) 獲得超6個(gè)贊
files = np.load("particle.npz")
x_s = [files[key] for key in files.keys()]
創(chuàng)建數(shù)組列表而不是單獨(dú)命名的數(shù)組是 Python 中的首選方法。
pox_s = [x[:,0] for x in x_s]
看起來數(shù)組的大小都相同。所以我們可以把列表變成數(shù)組：
pox_s = np.array(pox_s)
甚至
x_s = np.array(x_s) # (6, 40002, ?)
pox_s = x_s[:,:,0] # (6, 40002)
sq_diffs = (pox_s - pox_s[:,[0]])**2 # (6, 40002)-(6,1)
沒有一個(gè)具體的小例子，我無法測(cè)試這段代碼。我想我的形狀是對(duì)的。
滄海一幻覺
TA貢獻(xiàn)1824條經(jīng)驗(yàn) 獲得超5個(gè)贊
您基本上需要編寫一個(gè)額外的循環(huán)來遍歷字典的鍵。
files = np.load("particle.npz")
sq_diff_list = []
temp = []
for i in list(files.keys()):
arr = files[i]
for x in arr:
temp.append((x-arr[0])**2)
sq_diff_list.append(temp)
temp = []
添加回答
舉報(bào)
0/150
提交
取消