如何绘制TF.Dataset类标签的直方图

Question

我用TF.CsvDataset从磁盘中加载数据。将数据绘制为

#This is the transformation function applied on loaded data before displaying histogram.
def preprocess(*fields):
    print(len(fields))
    features=tf.stack(fields[:-1])
    labels=tf.stack([int(x) for x in fields[-1:]])
    return features,labels  # x, y

for features,label in train_ds.take(1000):
#  print(features[0])
plt.hist(features.numpy().flatten(), bins = 101)

我得到了这个直方图

​

​

但是，我想根据二进制类标签绘制712个特性值的分布图。也就是说，当类标签为0时，特性1、2或3的值是多少。

如何用pyplot实现这一点？

我读过下面的文章，但是，没有任何帮助。

绘制大熊猫/ matplotlib中类的直方图

直方图颜色按类别划分

如何在python中绘制多个类别的直方图

Answer 1

您可以使用np.fromiter并获取所有标签。然后您只需将标签列表传递给plt.hist。

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt

train, test = tf.keras.datasets.mnist.load_data()

ds = tf.data.Dataset.from_tensor_slices(train)

vals = np.fromiter(ds.map(lambda x, y: y), float)

plt.hist(vals)
plt.xticks(range(10))
plt.title('Label Frequency')
plt.show()

​

​