我有一個 NumPy 陣列,如果是奇數,我需要將值更改為 0,如果是偶數,我需要將值更改為 1。我怎樣才能做到這一點?
arr = np.array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])
uj5u.com熱心網友回復:
模運算子在這里很有用:
arr % 2
這導致1您在任何地方都有一個奇數(因為在整數除以 2 的情況下余數為 1),這與您想要的相反。
要翻轉它,請從 1 中減去它:
1 - arr % 2
根據您的資料,這給出:
array([[0, 1, 0, 1, 0],
[1, 0, 1, 0, 1]])
請注意,這實際上并沒有改變您的輸入資料,這是您所要求的。您可以通過覆寫arr此結果來實作。
轉載請註明出處,本文鏈接:https://www.uj5u.com/caozuo/518101.html
標籤:Python麻木的
