【急】怎么获取布尔列表True值的索引号？

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

如题：怎么杨获取下列列表中True值的索引号

1. tmplist = [True, True, False, True, False]

复制代码

  我想得到[0,1,3]的结果

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：Rue TRU list LSE tmp

1. tmplist = [True, True, False, True, False]
   
2. idx = []
   
3. for i in range(len(tmplist)):
   
4.     if tmplist[i] == True:
   
5.         idx.append(i)
   
6.         
   
7. print(idx)

复制代码

1. tmplist = [True, True, False, True, False]
   
2. print([i for i in range(len(tmplist)) if tmplist[i] == True])

复制代码

听雨在小楼 发表于 2022-4-10 10:15

谢谢！

pythonic solution should be as followint:

from itertools import compress
mask = range(5)
tmplist = [True, True, False, True, False]
print(list(compress(mask, tmplist))

pythonic way to solve this problem:

from itertools import compress
tmplist = [True, True, False, True, False]
mask = range(5)
print(list(compress(mask, tmplist)))

1. from itertools import compress
   
2. tmplist = [True, True, False, True, False]
   
3. mask = []
   
4. for i in range(5):
   
5.     mask.append([i for i in range(len(tmplist)) if tmplist[i] == True])
   
6. print(list(compress(mask, tmplist)))

复制代码

要的是下面的结果？

1. [[0, 1, 3], [0, 1, 3], [0, 1, 3]]

复制代码