我的dataframe 有一个列 data['ai_te'] 里面的值为{"red":0,"orange":0,"green":0,"white":0,"blue":0,"black":0,"yellow":0,"purple":0,"cyan":0,"grey":1} ,请帮我提取出为1,的key
时间: 2023-07-02 18:13:01 浏览: 81
可以使用 Python 中的 json 库来解析字符串,然后提取值为 1 的 key。具体代码如下:
```python
import json
# 假设数据存储在名为 data 的 DataFrame 中
# 首先将字符串转换为字典
data['ai_te'] = data['ai_te'].apply(lambda x: json.loads(x))
# 提取值为 1 的 key
data['ai_te'] = data['ai_te'].apply(lambda x: [k for k, v in x.items() if v == 1])
```
其中,`json.loads()` 可以将字符串转换为字典,`apply()` 方法可以将操作应用到整个 DataFrame 中的每一行。最后得到的 `data['ai_te']` 就是值为 1 的 key 的列表。
阅读全文