Почему выравнивание индекса так странно в этом кадре данных?Как это исправить? - PullRequest
0 голосов
/ 04 мая 2019

Хорошо, так что я работаю над этой проблемой ML, в которой я должен предсказать вероятность того, что человеку будет выдан штрафной билет. Я пытался установить ticket_id в качестве индекса данных, но произошло нечто странное, и я не знаю, почему.

import pandas as pd
import numpy as np

def blight_model():
    train = pd.read_csv('train.csv', encoding = "ISO-8859-1")
    test = pd.read_csv('readonly/test.csv', encoding = "ISO-8859-1")
    address = pd.read_csv('readonly/addresses.csv', encoding = "ISO-8859-1")
    """X = data.iloc[:,0:33]  #independent columns
    y = data.iloc[:,-1]    #target column i.e price range"""
    common_cols_to_drop = ['agency_name', 'inspector_name', 'mailing_address_str_number',
                           'violator_name', 'violation_street_number', 'violation_street_name',
                           'mailing_address_str_name', 'admin_fee', 'violation_zip_code',
                           'state_fee', 'late_fee', 'ticket_issued_date', 'hearing_date', 'violation_description',
                           'fine_amount', 'clean_up_cost', 'disposition', 'grafitti_status',
                           'violation_code', 'city']
    train_cols_to_drop = ['payment_status', 'payment_date', 'balance_due', 'payment_amount','compliance_detail', 'collection_status'] + common_cols_to_drop
    train = train.drop(train_cols_to_drop, axis=1).set_index('ticket_id')
    train = train[np.isfinite(train['compliance'])]

    return train.head()

Результат, который я получаю, таков? Что с ticket_id?

Imgur

1 Ответ

1 голос
/ 04 мая 2019

Это просто отображается таким образом, чтобы вы знали, что это индекс.

Вы получаете это из-за этого (устанавливая это как индекс):

.set_index('ticket_id')

в этой строке:

train = train.drop(train_cols_to_drop, axis=1).set_index('ticket_id')
...