修改2d numpy數組中分隔符后的行中的所有元素

如果我有一個2d numpy（整數，d也是整數）數組，如下

[[0 1 2 d]
 [3 4 d 5]
 [6 d 7 8]]

如何將每行d（包括d）之后的所有元素(row-wise)清零？

我使用了for循環，但我想知道是否有通過numpy的矢量化方法。我剛剛看到，也許你可以減去一個triangular矩陣：

[[0 0 0 d]
 [0 0 d 5]
 [0 d 7 8]]

但這并不能解決我的問題，因為它需要將d之前的值清零。

EDIT:

repr(array)

array([[0, 1, 2, d],
   [3, 4, d, 5],
   [6, d, 7, 8]], dtype=int64)

發布于 1 月前

? 最佳回答：

IIUC，你可以用cumsum制作一個掩碼，并用它來用where掩碼前導值：

d = 9
a = np.array([[0, 1, 2, d],
              [3, 4, d, 5],
              [6, d, 7, 8]])

out = np.where(np.cumsum(a == d, axis=1), a, 0)

具有cumprod和就地修改的變體：

a[np.cumprod(a!=d, axis=1).astype(bool)] = 0

Output:

array([[0, 0, 0, 9],
       [0, 0, 9, 5],
       [0, 9, 7, 8]])

中間掩碼（對于第一種方法，考慮non-zero個值True）：

# np.cumsum(a == d, axis=1)
array([[0, 0, 0, 1],
       [0, 0, 1, 1],
       [0, 1, 1, 1]])

# np.cumprod(a!=d, axis=1).astype(bool)
array([[ True,  True,  True, False],
       [ True,  True, False, False],
       [ True, False, False, False]])

公眾號：1024技術圈

?? 提供互聯網知識和資訊，分享IT前沿技術，熱門資源，大廠面試題 ??

修改2d numpy數組中分隔符后的行中的所有元素

熱門問答

pandas dataframe.to_sql在寫入MariaDB時拋出錯誤

JS：克隆數組并使用新引用獲得相同結果的最佳方法是什么？

awk gsub命令，變量中的語句為“arg”

刪除具有特殊字符的數據庫

提取數組元素

帶有哈希模式的Vue路由器無法在Github Pages上工作

如何在Python中遵守robots.txt規則進行網頁抓取

更新到macOS Catalina后，我的Xcode項目出現問題，該怎么辦

閉包和立即執行函數表達式（IIFE）在JavaScript中有什么區別

學習stm32hal庫什么樣的書籍是既適合入門又有體系的

修改注冊表中profilelist出現用戶文件不見了

TypeError: Cannot convert undefined or null to object

修改2d numpy數組中分隔符后的行中的所有元素

熱門問答

pandas dataframe.to_sql在寫入MariaDB時拋出錯誤

JS：克隆數組并使用新引用獲得相同結果的最佳方法是什么？

awk gsub命令，變量中的語句為“arg”

刪除具有特殊字符的數據庫

提取數組元素

帶有哈希模式的Vue路由器無法在Github Pages上工作

如何在Python中遵守robots.txt規則進行網頁抓取

更新到macOS Catalina后，我的Xcode項目出現問題，該怎么辦

閉包和立即執行函數表達式（IIFE）在JavaScript中有什么區別

學習stm32hal庫什么樣的書籍是既適合入門又有體系的

修改注冊表中profilelist出現用戶文件不見了

TypeError: Cannot convert undefined or null to object

JS：克隆數組并使用新引用獲得相同結果的最佳方法是什么？

awk gsub命令，變量中的語句為“arg”

更新到macOS Catalina后，我的Xcode項目出現問題，該怎么辦