我有一個(gè)名為的矩陣xs：array([[1, 1, 1, 1, 1, 0, 1, 0, 0, 2, 1], [2, 1, 0, 0, 0, 1, 2, 1, 1, 2, 2]])現(xiàn)在，我想用同一行中最近的前一個(gè)元素替換零（假定第一列必須為非零。）。粗略的解決方案如下：In [55]: row, col = xs.shapeIn [56]: for r in xrange(row): ....: for c in xrange(col): ....: if xs[r, c] == 0: ....: xs[r, c] = xs[r, c-1] ....: In [57]: xsOut[57]: array([[1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 1], [2, 1, 1, 1, 1, 1, 2, 1, 1, 2, 2]])任何幫助將不勝感激。
3 回答
天涯盡頭無女友
TA貢獻(xiàn)1831條經(jīng)驗(yàn) 獲得超9個(gè)贊
如果可以使用pandas，replace則將在一條指令中顯式顯示替換項(xiàng)：
import pandas as pd
import numpy as np
a = np.array([[1, 1, 1, 1, 1, 0, 1, 0, 0, 2, 1],
[2, 1, 0, 0, 0, 1, 2, 1, 1, 2, 2]])
df = pd.DataFrame(a, dtype=np.float64)
df.replace(0, method='pad', axis=1)
幕布斯7119047
TA貢獻(xiàn)1794條經(jīng)驗(yàn) 獲得超8個(gè)贊
我的版本基于逐步滾動(dòng)和初始數(shù)組的屏蔽，不需要其他庫(kù)（numpy除外）：
import numpy as np
a = np.array([[1, 1, 1, 1, 1, 0, 1, 0, 0, 2, 1],
[2, 1, 0, 0, 0, 1, 2, 1, 1, 2, 2]])
for i in xrange(a.shape[1]):
a[a == 0] = np.roll(a,i)[a == 0]
if not (a == 0).any(): # when all of zeros
break # are filled
print a
## [[1 1 1 1 1 1 1 1 1 2 1]
## [2 1 1 1 1 1 2 1 1 2 2]]
添加回答
舉報(bào)
0/150
提交
取消