如何将pandas isin输出数据帧转换为json或list格式

将pandas isin输出数据帧转换为JSON或List格式的方法有多种。下面我将介绍两种常用的转换方式。

方法一：使用pandas库的to_json()方法

首先，使用pandas的isin()方法筛选出符合条件的数据帧。
然后，使用to_json()方法将筛选后的数据帧转换为JSON格式的字符串。
最后，如果需要将JSON字符串转换为List格式，可以使用json库的loads()方法进行转换。

示例代码如下：

import pandas as pd
import json

# 创建示例数据帧
data = {'col1': ['A', 'B', 'C', 'D', 'E'],
        'col2': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)

# 使用isin()方法筛选数据帧
filtered_df = df[df['col1'].isin(['A', 'C', 'E'])]

# 将筛选后的数据帧转换为JSON格式的字符串
json_str = filtered_df.to_json(orient='records')

# 将JSON字符串转换为List格式
result_list = json.loads(json_str)

print(result_list)

方法二：使用pandas库的to_dict()方法

首先，使用pandas的isin()方法筛选出符合条件的数据帧。
然后，使用to_dict()方法将筛选后的数据帧转换为字典格式。
最后，如果需要将字典格式转换为List格式，可以直接使用字典的values()方法获取值的列表。

示例代码如下：

import pandas as pd

# 创建示例数据帧
data = {'col1': ['A', 'B', 'C', 'D', 'E'],
        'col2': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)

# 使用isin()方法筛选数据帧
filtered_df = df[df['col1'].isin(['A', 'C', 'E'])]

# 将筛选后的数据帧转换为字典格式
result_dict = filtered_df.to_dict(orient='records')

# 将字典格式转换为List格式
result_list = [list(item.values()) for item in result_dict]

print(result_list)

以上是两种常用的将pandas isin输出数据帧转换为JSON或List格式的方法。根据具体需求选择适合的方法进行转换。