Используйте метод или оператор if для вставки в переменную экземпляра (Python)

Question

Я пытаюсь найти более простой / короткий способ добавить к указанной c переменной экземпляра. Мне было трудно понять, что даже в Google, так что вот иллюстрация проблемы:

Class Person:
      def __init__(self):
            self.year2018_sample_a = 0
            self.year2018_sample_b = 0
            self.year2019_sample_a = 0
            self.year2019_sample_b = 0
            self.year2020_sample_a = 0
            self.year2020_sample_b = 0

    #This works but not really ideal
    #we get the year from data, but didnt write the whole code

      def add_to_year(self...):
            if year == 2018 and sample == 'a':
               self.year2018_sample_a += 1
            elif year == 2018 and sample == 'b':
               self.year2018_sample_b += 1
            elif year == 2019 and sample == 'a':
               self.year2019_sample_a += 1
            etc......

Есть ли возможность написать это без необходимости писать каждый год дважды? Идея ниже не работает как есть, потому что это просто строка. Но любые идеи были бы хороши.

      Pseudocode ideas:

      def add_to_year(..):
           datayear = get_from_data_column1
           datasample = get_from_data_column2

           self.f'year{datayear}_sample_{datasample}' += 1  -------This is the part where im 
                                                                  struggling to insert into changing 
                                                                  instance variables

Answer 1

Class Person:
    def __init__(self):
        self.samples = { year: { sample: 0 for sample in ('a', 'b') } for year in (2017,2018,2019) }

    def add_to(self, year, sample):
        self.samples[year][sample] += 1

instance = Person()
instance.add_to(2017, 'b')

Answer 2

Вы можете использовать getattr и setattr:

class Foo():
    def __init__(self):
        for i in range(10):
            setattr(self, f'year_{2000 + i}', i)
f = Foo()
for i in range(10):
    print(getattr(f, f'year_{2000 + i}'))
print(f"Year 2005: {f.year_2005}")

Выход:

0
1
2
3
4
5
6
7
8
9
Year 2005: 5