Найти сумму элементов строки, где индекс содержит строку

Question

В python у меня есть фрейм данных panda с именем data. У нас есть строка для каждого индекса. Как найти сумму значений в третьем столбце для всех строк, индекс которых содержит ключевое слово «ключ»?

Я могу сделать это с помощью цикла for, но это не умно:

count = 0
  for i in range(1,10):     
    if 'key' in data.index[i]:
       count += data.iloc[i,3]

Answer 1

Используйте str.contains и подсчитайте количество True случаев:

count = data.loc[data.index.str.contains('key'), data.columns[3]].sum()