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]

猜你喜欢

转载自blog.csdn.net/lovetaozibaby/article/details/124606124