我一直在研究Python中的一個函數，該函數查找數組中從各自的索引到數組開頭的所有元素的總和。示例： 輸入：[2,14,17,36]輸出：[2, 14+2, 17+14+2, 36+17+14+2]這是代碼。import matplotlib.pyplot as pltimport numpy as nparr = []a = np.array([2, 0, 0, 4, 0, 1, 0, 4, 5, 5])def rolling_sum(x):? ? total = 0? ? values = []? ? for i,j in enumerate(x):? ? ? ? total = total+j?? ? ? ? values.append(total)? ? if total <= 2000000000:? ? ? ? arr.append(values)? ? ? ? return rolling_sum(values)? ? else:? ? ? ? return valuesrolling_sum(a)for i in arr:? ? plt.plot(i)檢查arr變量發現其中有負數，甚至從圖表中也清楚地顯示出來。請問為什么會這樣呢？
2 回答
喵喔喔
TA貢獻1735條經驗 獲得超5個贊
當您執行代碼時，請注意警告：
RuntimeWarning: overflow encountered in long_scalars
total = total+j
這是因為numpy有時默認為np.int32（取決于操作系統是 32/64 位并且安裝的 Python 解釋器是 32/64 位，顯然是其中較低的一個）。在這種情況下，大量數字會溢出，因此它們會換行為負數。
通過提供以下內容可以輕松解決此問題dtype=np.int64：
a = np.array([2, 0, 0, 4, 0, 1, 0, 4, 5, 5], dtype=np.int64)
這可以通過添加以下內容來確認：
from itertools import chain
print(len(list(filter(lambda n: n < 0, chain.from_iterable(arr)))))
這會展平arr并計算有多少個負數。使用原始代碼，輸出是
RuntimeWarning: overflow encountered in long_scalars
total = total+j
5
添加后dtype=np.int64輸出是
0
胡子哥哥
TA貢獻1825條經驗 獲得超6個贊
NumPy 數組使用固定大小的整數類型（例如）?，這與大小不受限制的int64常規 Python 類型不同。int
它們具有可以代表的最大值。嘗試添加大于此最大值的值會導致溢出，這可能會導致有符號整數類型出現負值。
例如，該類型的最大值int32為 2?31??1 = 2147483647，加 1 得到最小值 ?2?31?= ?2147483648。對于int64類型來說，這些值要大得多；分別為9223372036854775807和-9223372036854775808。根據您的實現（即 NumPy 默認使用的類型）和您使用的輸入，您可能會或可能不會看到此行為。
就您而言，與 Python 列表相比，您似乎沒有使用 NumPy 數組提供的任何功能，因此您可以只使用a = [2, 0, 0, 4, 0, 1, 0, 4, 5, 5]并依賴 Python 為您處理無限的大整數。
添加回答
舉報
0/150
提交
取消