ML задание 3

# Я хорошо умею решать проблемы - Q11
# Q9 - 4, Q3 - 2, Q6 - 4
# ----------------------------------------------
# Я считаю, что у меня хорошо получается генерировать идеи - Q12
# Q1 - 3, Q8 - 4, Q10 - 3
# ----------------------------------------------
# Я стараюсь становиться лучше в моей специализации - Q13
# Q5 - 4, Q7 - 5, Q10 - 1
# ----------------------------------------------
import pandas as pd
import numpy as np

dataset_array = pd.read_csv(
    'dataset2.csv',
    delimiter=',',
    names=['NameOfUser', 'Q1', 'Q2',
           'Q3', 'Q4', 'Q5', 'Q6', 'Q7', 'Q8',
           'Q9', 'Q10']
)
Q11 = [0]*len(dataset_array)
Q12 = [0]*len(dataset_array)
Q13 = [0]*len(dataset_array)
for i in range(1, len(dataset_array)):
    answer = str((4 * float(dataset_array['Q9'][i]) + 2 * float(dataset_array['Q3'][i]) + 4 * float(dataset_array['Q6'][i])) / 10)
    Q11[i] = answer[0:8]
    answer = str((3 * float(dataset_array['Q1'][i]) + 4 * float(dataset_array['Q8'][i]) + 3 * float(dataset_array['Q10'][i])) / 10)
    Q12[i] = answer[0:8]
    answer = str((4 * float(dataset_array['Q5'][i]) + 5 * float(dataset_array['Q7'][i]) + 1 * float(dataset_array['Q10'][i])) / 10)
    Q13[i] = answer[0:8]
dataset_array['Q11'] = Q11
dataset_array['Q12'] = Q12
dataset_array['Q13'] = Q13

print(dataset_array)
dataset_array.to_csv('dataset3.csv', index=False)