from sklearn import datasets
import pandas as pd
import matplotlib.pyplot as plt
from numpy import array

iris = datasets.load_iris()

df = pd.DataFrame(iris.data, columns=iris.feature_names)
colors = array(50*['r']+50*['g']+50*['b'])
_ = pd.plotting.scatter_matrix(df, c=colors, figsize=[8,8], s=100, alpha=0.8)
plt.figure()

plt.imshow(X = [np.unique(iris.target)])
_ = plt.xticks(ticks=np.unique(iris.target),labels=iris.target_names)

我尝试将cmap参数添加到plt.imshow（），但如果X是RGB（A），则忽略此参数。这是文档Hyperlink中所写的。我尝试将X = colors改为，但它不起作用。