我正在嘗試優化此代碼：num = 10for j in xrange(0,num): u[j],v[j] = rk4(du,dv,t,dt,u[j],v[j])其中 u 和 v 是輸入數組， rk4() 為兩個輸入值返回兩個值。使用列表理解我會做這樣的事情：u,v=[rk4(du,dv,t,dt,u[j],v[j])) for j in range(0,num)]列表理解有效。但是輸出格式不同。是否可以使用列表理解來優化這種操作？編輯：所需的輸出將是表單的兩個數組/列表u,v = [u1,u2,u3,....],[v1,v2,v3,...]我得到的是以下形式：[(u1,v1),(u2,v2),(u3,v3),...]Python列表理解
1 回答
胡說叔叔
TA貢獻1804條經驗 獲得超8個贊
您似乎想將一對序列轉換為兩個序列。zip在 Python 中有一個標準的習慣用法是使用函數和參數解包來做到這一點：
>>> seq_of_pairs = [('a', 1), ('b', 2), ('c', 3), ('d', 4)]
>>> u, v = zip(*seq_of_pairs)
>>> u
('a', 'b', 'c', 'd')
>>> v
(1, 2, 3, 4)
因此，您可以使用列表推導（或生成器表達式）生成對序列zip，然后使用該技巧提取兩個序列：
result = [ rk4(..., ui, vi) for ui, vi in zip(u, v) ]
u, v = zip(*result)
u, v = map(list, zip(*result))如果您需要它們是列表而不是元組，您可以這樣做。
添加回答
舉報
0/150
提交
取消