[Python] Funktion som kan returnera index av en viss karaktär från en sträng?

Det är fritt fram att fråga gällande saker som var oklara om du undrar över något. Speciellt "list comprehensions" känns rätt magiska första gångerna man ser dem, men de beskriver ofta algoritmer på ett väldigt enkelt sätt, och bör användas där de passar.

Det kan kännas som onödigt mycket "jobb" att läsa in sig på hur en lösning fungerar när den väl gör vad du vill, men just för att det är ett så enkelt exempel så är det extra lärorikt att se till att man har full koll på vad som händer. I mer komplexa exempel så blir det mycket svårare att börja bena i koden.

Jag vet ingen direkt funktion i Pythons standardbibliotek som gör exakt det du frågar efter. Så som jag skrev min funktion så returnerar den en lista, ur vilken man sedan kan välja index enligt "vanlig" listsyntax.

Man skulle alternativt kunna skapa en funktion som tog ett ordningstal som ett extra argument och direkt returnerade ett index, men generellt är det trevligt att försöka skriva Python-kod som använder de mekanismer som språket erbjuder, vilket inkluderar []-konstruktionen för att välja listelement.

Om man letar efter den andra förekomsten av tecknet 'o' i strängen 'oslo' så skulle skillnaden mellan två sådana lösningar vara, om jag först upprepar den jag skrev ovan:

>>> def find_character_indices(haystack, needle):
...     return [i for i, character in enumerate(haystack) if character == needle]
... 
>>> find_character_indices('oslo', 'o')[1]
3

där vi använder listsyntaxen på den lista av träffar som returnerats. Nu den andra varianten:

>>> def find_character_index(haystack, needle, n):
...     return [i for i, character in enumerate(haystack) if character == needle][n - 1]
... 
>>> find_character_index('oslo', 'o', 2)
3

(notera den lilla skillnaden i funktionsnamn) där man alltså ger ordningstalet (2 för "den andra") för den träff man vill få indexet för.

Än enklare så skulle man kunna erbjuda båda alternativen och låta den andra varianten ärva funktion från den första:

>>> def find_character_indices(haystack, needle):
...     return [i for i, character in enumerate(haystack) if character == needle]
... 
>>> def find_character_index(haystack, needle, n):
...     return find_character_indices(haystack, needle)[n - 1]
... 
>>> find_character_index('oslo', 'o', 2)
3

Måhända överdriven omfaktorering här, men det kan ofta vara trevligt att låta varje funktion sköta "så lite som möjligt" på egen hand, så att de alla är fullkomligt självklar att beskriva (och testa!). När man skriver på detta sätt så ser man också tydligt att den andra funktionen, vars enda uppgift är att plocka ut ett listindex, känns rätt blek och att man troligen klarar sig utan den. Koden i den första funktionen känns vettigare att ha i en återanvändbar funktion.

Med den mer specifika beskrivningen så kan du enklare titta på split, eller kanske än mer splitlines:

>>> 'Never gonna give you up\nNever gonna let you down\nNever gonna run around and desert you'.splitlines()
['Never gonna give you up', 'Never gonna let you down', 'Never gonna run around and desert you']

Ifall det du läser "strömmas" från exempelvis en fil så finns det mer direkta sätt att läsa rad för rad. Om vi antar att vi har en fil astley med samma text som teststrängen ovan så kan du antingen läsa in hela filen i en lista där varje rad blir ett element med

>>> with open('astley') as f:
...     lines = f.readlines()
... 
>>> lines
['Never gonna give you up\n', 'Never gonna let you down\n', 'Never gonna run around and desert you\n']

(notera att radbrytningen är en del av varje element; se nedan för ett sätt att strippa den) eller så kan du läsa filen sekventiellt och agera på varje rad efterhand:

>>> with open('astley') as f:
...     for line in f:
...         print('[RA]: {}'.format(line.rstrip('\n')))
... 
[RA]: Never gonna give you up
[RA]: Never gonna let you down
[RA]: Never gonna run around and desert you

där jag använde rstrip för att rensa bort den avslutande extra radbrytningen för varje rad (beroende på exakt vad man vill göra kanske man vill sköta detta annorlunda).