porównywanie list zagnieżdżonych

Question

Cześć. Potrzebuje waszej pomocy. Mam 2 listy i chcę je porównać, a wynik zapisać do 3 listy. Wynikiem ma być lista obecności. Nie mogę sobie poradzić, ze stworzeniem pętli, która będzie się odwoływać do odpowiednich elementów listy. Proszę o pomoc.

pracownicy = [['001', 'Adam', 'Spadam'],
              ['002', 'Jan', 'Brat'],
              ['003', 'Marek', 'Sucharek'],
              ['004', 'Jarek', 'Barek'],
              ['005', 'Mirek', 'Wampirek']]


obecni = ['001', '004', '005', '008']


lista_obecnosci = []


for i in obecni:
        if i in pracownicy[][0]:
            lista_obecnosci.append([pracownicy[][1], pracownicy[][2]])
        else:
            lista_obecnosci.append([pracownicy[][0], '-niezidentyfikowany'])


plik = open('plik.txt', 'w')
plik.writelines(lista_obecnosci)
plik.close()

 

Answer 1

Proponuję użyć słownika do przechowywania pracowników. https://docs.python.org/2/tutorial/datastructures.html#dictionaries

pracownicy = {'001': ['Adam', 'Spadam'],
              '002': ['Jan', 'Brat'],
              '003': ['Marek', 'Sucharek'],
              '004': ['Jarek', 'Barek'],
              '005': ['Mirek', 'Wampirek']}
 
 
obecni = ['001', '004', '005', '008']
 
lista_obecnosci = []

for id in obecni:
  if id not in pracownicy:
    continue
  pracownik = pracownicy[id]
  lista_obecnosci.append(pracownik)

 

Comments

Dzięki, działa! Ale to tworzy mi problem we wcześniejszej części programu, otóż ta lista z pracownikami pobierana była z pliku tekstowego. Da się to przerobić, żeby działało do słownika?

pracownicy = []

text_file = open("baza_pracownikow.txt", "r")
for line in text_file:
        line = line.replace("\n", "")
        line = line.replace("\r", "")
        pracownicy.append(line.split(" "))
text_file.close()

Plik tekstowy z bazą pracowników ma taki układ:

001 Adam Spadam
002 Jan Brat
003 Marek Sucharek

 

 

---

Tak naprawdę to nie znam się na pythonie.
Myślę że coś takiego by dało radę:

pracownicy = {}
 
text_file = open("baza_pracownikow.txt", "r")
for line in text_file:
        line = line.replace("\n", "")
        line = line.replace("\r", "")
        dane = line.split(" ")
        id = dane.pop(0)
        pracownicy[id] = dane
text_file.close()

Źródła z których korzystałem:
https://stackoverflow.com/questions/4426663/how-to-remove-the-first-item-from-a-list
https://www.journaldev.com/23232/python-add-to-dictionary

---

Tak, tego potrzebowałem. Działa świetnie!