Translate a normal for loop structure into list comprehension structure

Question

I am trying to translate the below structure:

newDV = []

for row in dataVector:

   for cell in row:

      newDV.append((cell if row.index(cell) != 0 else 'other'))

This is the list comprehension structure:

[

    cell

    for cell in row

    for row in dataVector

    if row.index(cell) != 0 else 'other'

]

I am getting the error:

UnboundLocalError: local variable 'row' referenced before assignment

Answer 1

Try this:

[

    cell

    for row in dataVector

    for cell in row

    if row.index(cell) != 0 else 'other'

]

 If you are a beginner and want to know more about Data Science the do check out the Data Science course