从两个数字 pandas 列创建 Python 维恩图

Question

我有两个 pandas 列（下面是可重现的输出）

{'Test_actual': {0: 160.702, 1: 113.457, 2: 91.245, 3: 53.784, 4: 40.281, 5: 39.236, 6: 37.73, 
7: 32.692, 8: 29.983, 9: 29.983, 10: 29.69, 11: 29.69, 12: 26.232, 13: 25.779, 14: 24.094, 15: 
24.094, 16: 22.46, 17: 20.731, 18: 20.731, 19: 17.367}, 'Test_predicted': {0: 24.68385875, 1: 
21.22846956, 2: 46.41633486, 3: 0.196713859, 4: 18.22344042, 5: 13.87247076, 6: 28.5820542, 7: 
59.67648307, 8: 13.95858298, 9: 20.03071567, 10: 2.73188936, 11: 2.73188936, 12: 15.57722262, 
13: 9.469598881, 14: 0.311162267, 15: 38.29214566, 16: 11.47778436, 17: 8.754663155, 18: 
19.61416015, 19: 1.858514339}}

我正在尝试在 Python 中构建维恩图。尝试创建集，但不起作用。

重要。我想保留所有值，而不仅仅是唯一值。

import matplotlib.pyplot as plt
from matplotlib_venn import venn2

col_one_list = rf_test_actual_pred_10['Test_actual'].tolist()
col_two_list = rf_test_actual_pred_10['Test_predicted'].tolist()

set1 = set(col_one_list)
set2 = set(col_one_list)
venn2([set1, set2], ('Test actual', 'Test predicted'))

plt.show()

当前输出错误

预期输出

如果我没有准确的值，是否应该将重叠值四舍五入？

Answer 1

无需先将您的列转换为 list。直接使用 set:

试试

>>> venn2([set(rf_test_actual_pred_10["Test_actual"].astype(int)),
           set(rf_test_actual_pred_10["Test_predicted"].astype(int))],
           set_labels=("Test actual","Test predicted"))

从两个数字 pandas 列创建 Python 维恩图

Create a Venn diagram in Python from two numeric pandas columns

python

venn-diagram

pandas

matplotlib-venn

venn