根据条件语句从图中删除列

Question

我正在尝试调整 Python 中的相关矩阵，以便从数据框中，在给定我的函数参数的情况下，只绘制存在数据的列。问题是我认为根据我的代码，绘图会自动调整，并且没有数据的列不会显示在绘图中。因为它是一个相关矩阵，图中出现的列越多，相关性就越不可见和“缩小”。负责删除列和绘制矩阵的代码部分如下所示。这被嵌入到一个更大的功能中。评论的部分是我尝试选择没有数据的列的不同方式，但都没有用。我是通过原始数据框的副本完成的，因为我不想从那个数据框中删除列。当然seaborn已经导入了

dfCorr2 = dfCorr
for col in dfCorr2:
    #   if dfCorr[column].count(axis=1) > 0:
    #   if column.notnull().count() > 0:
    #   if dfCorr.isnull().sum(axis=1) > 0:
    if dfCorr2[col].isna().sum() > 0:
    print("The column(s) with no data for this trial are: ", column)
    dfCorr2 = dfCorr2.drop(col)


correlation_mat = dfCorr2.corr()
sns.set(rc={'figure.figsize':(11.7,8.27)})
sns.heatmap(correlation_mat, annot = True)

我希望这个数字能适应所选的列，而不是显示已删除的列。