如何解决关于列表理解的多个条件语句
这是我的代码,我想知道我是否可以使用列表理解来执行相同的操作(计算行内的簇数并输出长度为 env.policies['json.dumps_kwargs'] = {'sort_keys': False}
的列表)。同一个簇号至少有两行,但可以更多,而且它们会循环。我试过了,但无法弄清楚。
有什么建议吗?
我的代码:
df.shape[0]
DataFrame 看起来像
索引 | 集群 |
---|---|
0 | 0 |
1 | 0 |
2 | 1 |
3 | 1 |
4 | 1 |
5 | 2 |
6 | 2 |
7 | 0 |
8 | 0 |
... | ... |
n | 米 |
解决方法
你想要这个吗?
from itertools import groupby
result = [0 if index == 0 and key == 0
else index
for index,(key,group) in enumerate(groupby(my_values))
for _ in group
]
print(result)
通过 - df['cluster'].values 替换列表推导式中的 my_values。测试
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。