如何循环下面的语句

Question

sorted(X_train_num.F12.unique())==sorted(X_test_num.F12.unique())

基本上我想在不同的变量上运行循环，如F12，F11，F78等，这样我就可以避免一次又一次地为不同的变量编写单个命令。

有人知道用不同的var（F12，F11等）

循环这个语句的简单方法

Answer 1

使用getattr：

var = 'F12'
sorted(getattr(X_train_num, var).unique())==sorted(getattr(X_test_num, var).unique())

以上内容可以轻松放入循环中。例如：

for var in ('F10', 'F11', 'F12'):
    sorted(getattr(X_train_num, var).unique())==sorted(getattr(X_test_num, var).unique())

Answer 2

附注：您还可以使用 set 相等来更有效地进行比较：

set(getattr(X_train_num, var)) == set(getattr(X_test_num, var))