Python将值映射到每个第i个子列表的元素

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了Python将值映射到每个第i个子列表的元素，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1183字，纯文字阅读大概需要2分钟。

内容图文

我正在尝试在python中执行以下操作：给出列表和整数i

input = [[1, 2, 3, 4], [1, 2, 3, 4], [5, 6, 7, 8]]
i = 1

我需要获得另一个列表,其中包含第i个列表元素的全1,否则为0

output = [0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0]

我写了这段代码

output = []
for sublist in range(0, len(input)):
    for item in range(0, len(input[sublist])):
        output.append(1 if sublist == i else 0)

它显然有效,但由于我是python的新手,我想有一种更好的“pythonic”方式.

我认为使用map可以工作,但我无法用它获取列表的索引.

解决方法:

创建额外的变量来获取当前元素的索引是非常不合理的.通常的替代方案是使用enumerate内置功能.

Return an enumerate object. sequence must be a sequence, an iterator,
or some other object which supports iteration. The next() method of
the iterator returned by enumerate() returns a tuple containing a
count (from start which defaults to 0) and the values obtained from
iterating over sequence.

您可以在其中使用带有双循环的列表理解来简化一行：

input_seq = [[1, 2, 3, 4], [1, 2, 3, 4], [5, 6, 7, 8]]
i = 1
o = [1 if idx == i else 0 for idx, l in enumerate(input_seq) for _ in l]

或者,

o = [int(idx == i) for idx, l in enumerate(input_seq) for _ in l]

Underscore只是一次性名称,因为在这种情况下,我们不关心存储在输入子列表中的实际值.

内容总结

以上是互联网集市为您收集整理的Python将值映射到每个第i个子列表的元素全部内容，希望文章能够帮你解决Python将值映射到每个第i个子列表的元素所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端