擅长:python、mysql、java
<p>代码设置:</p>
<pre><code>import pandas as pd
import numpy as np
test_matrix = np.array([[np.nan,10,20],[30,np.nan,40],[50,60,np.nan]])
lower_triangular = test_matrix[np.tril_indices(test_matrix.shape[0], -1)]
lower_triangular = list(np.tril(test_matrix).flat)
upper_triangular = list(np.triu(test_matrix).flat)
lower_triangular = [x for x in lower_triangular if x>0]
upper_triangular = [x for x in upper_triangular if x>0]
cus1 = np.tril_indices(test_matrix.shape[0], -1)[0]
cus2 = np.tril_indices(test_matrix.shape[0], -1)[1]
q = pd.DataFrame(
{'cus1': cus1,
'cus2': cus2,
'lower': lower_triangular,
'upper': upper_triangular
})
</code></pre>
<p>输出:</p>
<pre><code> cus1 cus2 lower upper
0 1 0 30.0 10.0
1 2 0 50.0 20.0
2 2 1 60.0 40.0
</code></pre>