Return list of index for pandas header list with duplicate values

Question

I’m trying to generate a unique index list with a for loop and a condition. It seems that because my original list has duplicate values, it’s only returning the first found index multiple times.

my list: ['Section', 'FINANCE Cost Category', 'Major Tasks', 'Sub Tasks', 'Start Date', 'End Date', 'Resource Code', 'Resource', 'Reporting Region', 'Rate', 'Effort', nan, 'Unit of Measure (UOM)', 'Invoice Trigger', 'Unit Hours', 'Total FTE', 'Unit Cost', '# of Units ', 'Extended Hours', 'Budget Total', '# of Units ', 'Extended Hours', 'Budget Total', nan, '# of Units ', 'Extended Hours', 'Budget Total', nan, '# of Units ', 'Extended Hours', 'Budget Total', nan, '# of Units ', 'Extended Hours', 'Budget Total', nan, '# of Units ', 'Extended Hours', 'Budget Total', nan, '# of Units ', 'Extended Hours', 'Budget Total', nan, '# of Units ', 'Unit Hours', 'Unit Cost', 'Extended Hours', 'Budget Total', 'Service Provider Comments', nan, nan, nan, nan, nan, nan, nan]

My block:

header_mod_list = []
for each in headers:
    if 'budget' in str(each).lower():
        header_mod_list.append(headers.index(each))

print(header_mod_list)

This is yielding:

[19, 19, 19, 19, 19, 19, 19, 19]

What I want:

[19, 22, 26, 30, 34, 38, 42, 48]

Any thoughts?

Asked By: Mike Mann

||

Source

Answer 1

Use enumerate:

header_mod_list = []
for i, each in enumerate(headers):
    if 'budget' in str(each).lower():
        header_mod_list.append(i)

print(header_mod_list)

Answered By: Colin

Return list of index for pandas header list with duplicate values

Question:

Answers: