擅长:python、mysql、java
<pre><code>df = pd.DataFrame(
{ 'A': [0, 0, 0],
'B': [1, 0, 0],
'C': [0, 1, 0],
'D': [0, 0, 0],
'E': [1, 0, 1],
'F': [0, 0, 1],
}
)
df
A B C D E F
0 0 1 0 0 1 0
1 0 0 1 0 0 0
2 0 0 0 0 1 1
</code></pre>
<p>你可能在看一个像上面那样的测向仪。可以执行以下操作来拉取包含1的列。你知道吗</p>
<pre><code>columns = []
for col in df.T:
columns.append(df.T[df.T[col] == 1].index.tolist())
has1 = pd.DataFrame(columns).apply(lambda x: ', '.join(x[x.notnull()]), axis = 1)
df['Is1'] = has1
df
A B C D E F Is1
0 0 1 0 0 1 0 B, E
1 0 0 1 0 0 0 C
2 0 0 0 0 1 1 E, F
</code></pre>