将dataframe 数据中 {"red":0,"orange":0,"green":0,"white":0,"blue":0,"black":0,"yellow":0,"purple":0,"cyan":0,"grey":1} 提取出为1的颜色
时间: 2023-07-02 11:13:00 浏览: 128
拾取色彩值
你好!根据你的要求,可以使用 pandas 中的 DataFrame 进行操作。假设数据存储在名为 df 的 DataFrame 中,你可以使用以下代码提取出值为 1 的颜色:
```python
# 提取值为 1 的颜色
colors = df.columns[df.iloc[0].values == 1].tolist()
# 输出结果
print(colors)
```
其中,`df.columns` 是 DataFrame 中所有的列名,`df.iloc[0].values` 是 DataFrame 中第一行的所有值,我们通过条件 `df.iloc[0].values == 1` 来筛选出值等于 1 的元素。最后,使用 `.tolist()` 将结果转换为列表形式输出。
阅读全文