Чтение нескольких столбцов url cv файла

Question

Кто-нибудь знает, что не так с моим кодом? У меня возникли проблемы с нарезкой набора данных df в новый набор данных с двумя или более столбцами. Набор данных называется уровнем преступности в США, и я хотел бы создать новый набор данных с именем df1 только с столбцом изнасилования и убийства. Ilo c работает без каких-либо проблем, но я хотел бы разрезать набор данных, используя заголовок столбца, а не номер столбца. Вот мой код:

import pandas as pd
url= 'https://gist.githubusercontent.com/GeekOnAcid/da022affd36310c96cd4/raw/9c2ac2b033979fcf14a8d9b2e3e390a4bcc6f0e3/us_nr_of_crimes_1960_2014.csv'
dataset = pd.read_csv(url)
df = pd.DataFrame(dataset)
df1 = df[['Rape', 'Murder']]

Answer 1

Вы можете использовать метод rename для удаления пробелов из имен столбцов. Подробную информацию о методе rename можно прочитать в официальной документации .

import pandas as pd
url= 'https://gist.githubusercontent.com/GeekOnAcid/da022affd36310c96cd4/raw/9c2ac2b033979fcf14a8d9b2e3e390a4bcc6f0e3/us_nr_of_crimes_1960_2014.csv'
dataset = pd.read_csv(url)
df = pd.DataFrame(dataset)
df.rename(columns=lambda x:x.strip(), inplace=True)
print(df.columns)
df1 = df[['Rape', 'Murder']]
print(df1.head())

Вывод:

Index(['Year', 'Total', 'Violent', 'Property', 'Murder', 'Rape', 'Robbery',
       'Aggravated-Assault', 'Burglary', 'Larceny-Theft', 'Vehicle-Theft'],
      dtype='object')
   Rape  Murder
0   9.6     5.1
1   9.4     4.8
2   9.4     4.6
3   9.4     4.6
4  11.2     4.9