解释代码test_X_i = X_test[np.where(np.array(test_SNRs) == snr)]

这行代码的作用是从测试数据集中选取信噪比为 snr 的样本，并将其存储在 test_X_i 中。

具体来说，np.array(test_SNRs) == snr 会返回一个布尔数组，其中元素为 True 表示对应的样本的信噪比等于 snr。然后，np.where() 函数会返回所有 True 元素的下标，即对应的样本在测试数据集中的下标。最后，X_test[np.where(np.array(test_SNRs) == snr)] 会选取这些下标对应的样本，即信噪比为 snr 的样本，并将其存储在 test_X_i 中。