如何使用单个打印语句确定行索引?

时间:2018-12-09 00:24:03

标签: python arrays python-3.x dataframe

给出行索引的数据为max_sw,列表为sw_col

我尝试了这个以及其他一些变体,但是没有任何效果。

print(i for i in range(len(sw_col)) if sw_col[i]== max_sw)

1 个答案:

答案 0 :(得分:0)

您所拥有的电话线几乎在那如果将生成器放入列表中并且仅使用索引位置零,则将为您提供正确的答案:

sw_col = ['a','b','c']
max_sw = 'c'
print([i for i in range(len(sw_col)) if sw_col[i]== max_sw][0])  # prints 2

更简洁的解决方案是直接在列表中查找项目,如下所示:

sw_col = ['a','b','c']
max_sw = 'c'
print(sw_col.index(max_sw))  # prints 2