pisanie "pythonic" kodu

Question

Witam, robiłem zadanie na codewars https://www.codewars.com/kata/equal-sides-of-an-array/train/python
w skrócie polega na tym że funkcja ma zwrócić indeks listy dla którego suma liczb na lewo od niego jest równa sumie liczb na prawo od niego. Próbuje zrobić to w stylu pythona i napisałem coś takiego, ale zwraca mi synaxerror i nie mam pojęcia gdzie jest błąd, macie jakiś pomysł?

def find_even_index(arr):
    return i if sum(arr[:i]) == sum(arr[i+1:]) for i in range(len(arr))

Swoją drogą poleci ktoś jakiś tutorial do pythona który nie jest jednocześnie wprowadzeniem do programowania żeby nie tłumaczyli od nowa jak działa pętla itd.?

Comments

Swoją drogą poleci ktoś jakiś tutorial do pythona który nie jest jednocześnie wprowadzeniem do programowania żeby nie tłumaczyli od nowa jak działa pętla itd.?

Niestety mało jest takich, zwykle trzeba pominąć po prostu początkowe "rozdziały" / sekcje.

Answer 1

Brakuje Ci nawiasów. Kwadratowych dla list comprehension. Okrągłych dla generatora. W obu przypadkach zwracasz listę/generator, a nie pojedynczy indeks. Dodatkowo if musi wystąpić za instrukcją for. Tak wygląda całość bez syntax errorów:

def find_even_index(arr):
    return [i for i in range(len(arr)) if sum(arr[:i]) == sum(arr[i+1:])]

A w ten sposób wyciągniesz pierwszy element, o ile taki istnieje, w przeciwnym wypadku None

def find_even_index(arr):
    return next((i for i in range(len(arr)) if sum(arr[:i]) == sum(arr[i + 1 :])), None)