主页 / 问题 / 1038713
Accepted
GreyKoshak
Asked:2020-10-26 02:21:58 +0000 UTC2020-10-26 02:21:58 +0000 UTC

DataFrame：查找所有空元素的索引

  • 772

我正在解决一个 TSP（旅行推销员问题）。该算法提供了一个用于处理矩阵数据的循环 - 搜索所有零元素并顺序评估这些元素中的每一个以做出决定。这就是循环的来源。为了找到所有空元素的向量，我使用方法

result = np.where(df == 0)  # Найти все нулевые элементы - Numpy.
v_null = zip(result[0], result[1])  # Вектор, содержащий координаты нулевых элементов

DataFrame 中是否有类似的方法可以返回由某个条件找到的所有元素的索引向量？

python

1 个回答

  1. Best Answer

    有，例如：

    In [85]: df = pd.DataFrame(np.random.randint(0,5,size=(5, 5)), columns=list("abcde"))

In [86]: df
Out[86]:
   a  b  c  d  e
0  2  3  2  3  1
1  4  3  3  1  4
2  4  2  2  2  3
3  0  4  3  2  3
4  4  2  0  0  0

In [87]: df[df.eq(0)].stack().reset_index()
Out[87]:
   level_0 level_1    0
0        3       a  0.0
1        4       c  0.0
2        4       d  0.0
3        4       e  0.0

    列level_0和level_1- 包含空元素的索引。

    • 2

相关问题