python - 基于 pandas 系列列表创建二维绘图像素网格

Question

假设我们有一个 pandas 系列列表，其中每个列表包含一些描述为字符串的特征，如下所示:

0  ["A", "C", "G", ...]
1  ["B", "C", "H", ...]
2  ["A", "X"]
...
N  ["J", "K", ...]

什么是最好/最简单的绘制二维像素网格的方法，其中 X 轴是特征的出现，Y 轴是系列 0,1,2,..., N 中的每个样本？

16 月 22 日编辑:

我似乎没有明确提到特征列表不一定对所有观察具有相同的大小。观察 1 可以有 4 个特征，观察 2 可以没有一个，观察 3 可以有 5 个，依此类推。因此，如果不以某种方式对它们进行预处理以填充缺失的特征，我无法立即将它们转换为 numpy 数组。

Answer 1

由于我已经在我的评论中为图像编写了代码，而且 Ed 似乎和我对你的问题有相同的解释，我将继续并添加我的解决方案。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import string

M, N = 100, 10
letters = list(string.ascii_uppercase)
data = np.random.choice(letters, (M, N))

df = pd.DataFrame(data)
# Get frequency of letters in each column using pd.value_counts
df_freq = df.apply(pd.value_counts).T

# Plot frequency dataframe with seaborn heatmap
ax = sns.heatmap(df_freq, linewidths=0.1, annot=False, cbar=True)
plt.show()