Почему мое сравнение строк не работает в Python? ⇐ Python

[Anonymous]

Я написал простой код на Python на основе данных опроса

Код: Выделить всё

`
# Load CSV data
csv_file = 'women_survey.csv'
data_women = pd.read_csv(csv_file)

# Convert object to String
data_women['physical_Assault'] = data_women['physical_Assault'].astype("string")
w_marital_status = data_women['physical_Assault']

# print sample data size and type
print(w_marital_status.shape)
print(w_marital_status.dtype)
# person 1's data
print(w_marital_status[0])

# Create a new List
new_list = [""] * 62

# Read each iteration of survey data and assign String to new List
for count, content in enumerate(w_marital_status, start=0):
new_test = content.split(":", 1)[1]
new_list[count] = new_test
print(new_test)
if type(new_test) == str:
if new_test == "Yes":
print("This is a Yes")
else:
print("This is a No")
else:
print("This is not a string.")`

Этот код дает следующий результат (его часть).

Код: Выделить всё

(62,)
string
Have you ever been physically assaulted?: No

No

This is a No
No

This is a No
No

This is a No
No

This is a No
Yes

This is a No

Как показано в результате, для каждого человека каждая строка содержит весь вопрос «Подвергались ли вы когда-либо физическому насилию?» вместе с добавленным ответом «Да» или «Нет»
Я хочу сохранить только ответы «Да» или «Нет», чтобы позже использовать MatplotLib и построить график данных. поэтому я написал цикл For, чтобы назначить ответы новому списку. Но, как видите, сравнение строк никогда не работает. Я не был уверен, что "new_test" на самом деле является строкой. Оказывается, это строка, но сравнение строк все равно не работает. Я не совсем уверен, что я сделал не так. Также, если есть более простой способ, чем мой код, дайте мне знать. Спасибо.

Подробнее здесь: https://stackoverflow.com/questions/789 ... -in-python