python 用if選擇滿足指定條件的列

[col for col in data.columns if data_train[col].nunique() <= 1]

判斷data.columns的列中，if該列滿足條件，則將col加入列表中。