Длины чтения функции Python не работают

Я пытаюсь разработать функцию, которая проверяет длину VIN и возвращает результат.

Чтобы проверить, у меня есть распечатка вина, если длина = 17. Однако он четко распечатывает все. Не уверен, что я делаю неправильно.

Пример кода ниже.

import pandas as pd

#initialization
df = pd.DataFrame(columns = ["vin"], data = ['LHJLC79U58B001633','SZC84294845693987','LFGTCKPA665700387','L8YTCKPV49Y010001',
                                         'LJ4TCBPV27Y010217','LFGTCKPM481006270','LFGTCKPM581004253','LTBPN8J00DC003107',
                                         ƇA9LPEER3FC596536',ƇA9LREAR5FC596814',ƇA9LKEER2GC596611',ƇA9L0EAH9C596099',
                                         དྷA000018'])

df['manufacturer'] = ['A','A','A','A','B','B','B','B','B','C','C','D','D']

#develop function
def check_vin(df):
    if len(df['vin'][1]) == 17:
        print(df['vin'])
    else:
        print('nogo')


#test the function
for index, row in df.iterrows():
    check_vin(df)

Всего 4 ответа


Это потому, что вы просите длину одной и той же вещи на каждой итерации (df ['vin'] [1])

Изменить на

for index, row in df.iterrows():
   check_vin(row)

А также

def check_vin(r):
    if len(r.vin) == 17:
        print(r)
    else:
        print('nogo')

Выход

LHJLC79U58B001633
SZC84294845693987
LFGTCKPA665700387
L8YTCKPV49Y010001
LJ4TCBPV27Y010217
LFGTCKPM481006270
LFGTCKPM581004253
LTBPN8J00DC003107
1A9LPEER3FC596536
1A9LREAR5FC596814
1A9LKEER2GC596611
nogo
nogo

Не перебирайте строки вручную. Вы можете использовать назначить новый столбец с условной логикой:

df['check'] = np.where(df['vin'].str.len().eq(17), df['vin'], 'nogo')

print(df)

                  vin manufacturer              check
0   LHJLC79U58B001633            A  LHJLC79U58B001633
1   SZC84294845693987            A  SZC84294845693987
2   LFGTCKPA665700387            A  LFGTCKPA665700387
3   L8YTCKPV49Y010001            A  L8YTCKPV49Y010001
4   LJ4TCBPV27Y010217            B  LJ4TCBPV27Y010217
5   LFGTCKPM481006270            B  LFGTCKPM481006270
6   LFGTCKPM581004253            B  LFGTCKPM581004253
7   LTBPN8J00DC003107            B  LTBPN8J00DC003107
8   1A9LPEER3FC596536            B  1A9LPEER3FC596536
9   1A9LREAR5FC596814            C  1A9LREAR5FC596814
10  1A9LKEER2GC596611            C  1A9LKEER2GC596611
11   1A9L0EAH9C596099            D               nogo
12          22A000018            D               nogo

Вы можете просто найти свои 17 строк длины следующим образом:

condition = (df["vin"].str.len() == 17)
print(df[condition])

Все, что тебе нужно:

df[df.vin.str.len()==17]

Есть идеи?

10000