gpt4 book ai didi

获取索引的pythonic方式,值== 1的列

转载 作者:太空宇宙 更新时间:2023-11-04 08:54:31 24 4
gpt4 key购买 nike

我目前有一个 MxN 数据框,其中包含优化问题的解决方案。 {M} 中的 i 和 {N} 对中的“事件”i,j 由 1 表示,“非事件”对由 0 表示。我需要确定所有事件单元格的 i,j 值,希望避免 for 循环索引或列。

这是一个例子:

In [73]: sol_df
Out[73]:

1 2 3 4 5
1 0 0 1 0 0
2 1 0 0 0 0
3 0 1 0 0 0
4 0 0 0 0 0

在这种情况下,我需要的是一个成对列表(元组就可以):

[(1,3), (2,1), (3,2)]

有办法吗?

谢谢!

A.

编辑:解释不清楚EDIT2:我的解释仍然不清楚

最佳答案

>>> import numpy
>>> a = numpy.array([[1, 0, 1], [0, 1, 1], [0, 1, 0]])
>>> numpy.transpose(numpy.nonzero(a))
array([[0, 0],
[0, 2],
[1, 1],
[1, 2],
[2, 1]])

关于获取索引的pythonic方式,值== 1的列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31520952/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com