Python | 从列表或矩阵中选择某个符合条件的数据

简介

~~有一个神奇的事情，我看过很多人用a[a<3]这样的语法来筛选数据，但是每次我用的时候都会报错~

直到今天，我仔细调试了一下才发现，该语句是有适用范围的。
比如说：矩阵可以用，但列表不可以。

下面做几个小的演示。

测试程序

l = [1,2,3,4,5]  #测试列表
l[l<3] # 语句不会通过，会报错

l_n = np.array(l)
l_n[l_n<3]  # 输出：[1, 2]

总结

1. 矩阵适用该语句l_n[l_n<3]
2. 列表不适用

若要寻找列表中满足条件的数，可以用循环：

l = [1,2,3,4,5]  #测试列表
[i for i in l if i < 3] # 输出：[1,2]