擅长:python、mysql、java
<p>将<code>np.where</code>与<code>np.column_stack</code>一起使用:</p>
<pre><code>randValues = np.random.rand(5,5)
df = pd.DataFrame(randValues)
df_bool = df > 0.8
ind = np.column_stack(np.where(df_bool)))
print(ind)
colArray = [i[1] for i in ind] # [2,3]
rowArray = [i[0] for i in ind] # [0,1]
</code></pre>
<p>输出:</p>
<pre><code>array([0,2],
[1,3])
</code></pre>