ОБНОВЛЕНИЕ: ошибка в командной строке: ValueError: неверный путь к файлу или тип буферного объекта: - PullRequest
0 голосов
/ 27 мая 2020

Теперь моя функция работает следующим образом: несмотря на то, что она выполняет правильное решение, я получаю огромную ошибку в моей строке CMD https://i.stack.imgur.com/PBTwz.png (ValueError: неверный путь к файлу или тип буферного объекта: <класс 'pandas .core.frame.DataFrame'>). Кто-нибудь знает, почему я получаю эту ошибку, даже если вывод выполняется правильно?

import pandas as pd
import numpy as np
import argparse

target_col = "CountryRefs"

sep = ","

input_file = 'Test_set'

def arg_parse():
    parser = argparse.ArgumentParser()
    parser.add_argument("-f", "--input_file", required = True)
    parser.add_argument("-s", "--sep", required=True,)
    parser.add_argument("-t", "--target_col", required=True)
    args=parser.parse_args()
    return vars(args)

def splitter(input_file, target_col, sep, new_col = None, *argv):
    df = pd.read_csv(input_file)
    df[target_col] = df[target_col].str.split(sep)
    exploded = df.explode(target_col)
    exploded[target_col].replace(r'^\s*$', np.nan, regex=True, inplace = True)
    exploded.dropna(subset=[target_col], inplace=True)
    if new_col == None:
        return(pd.DataFrame(exploded[[target_col,*argv]]))
    else:
        exploded[new_col] = exploded[target_col]
        return(pd.DataFrame(exploded[[new_col,*argv]]))

if __name__ == '__main__':
    args = arg_parse()
    print(splitter(**args))

1 Ответ

1 голос
/ 27 мая 2020

Вам нужно что-то вроде этого

import pandas as pd
import numpy as np
import argparse


target_col = "CountryRefs"

sep = ","

data = {'CountryRefs':['Italy, Germany', 'Japan , France', '', 'Alaska'],
    'Authors':['Dom', 'Xavier', 'Kathleen', 'Joe'], 'Friends':['Amy Pete', 'Joe', None, 'Franklin'],
    'Colors':['red.blue', ' ', 'yellow', 'black.blue']}
df = pd.DataFrame(data, columns = ['CountryRefs', 'Authors', 'Friends', 'Colors'])

def arg_parse():
    parser = argparse.ArgumentParser()
    parser.add_argument('argv', type=str, nargs='*', default=[])
    parser.add_argument("-s", "--sep", dest="sep", required=True, default=',')
    parser.add_argument("-t", "--target_col", dest="target_col", required=True, default='1')

    args=parser.parse_args()
    return vars(args)

def splitter(df, target_col, sep, new_col = None, argv=[]):

    df[target_col] = df[target_col].str.split(sep)
    exploded = df.explode(target_col)
    exploded[target_col].replace(r'^\s*$', np.nan, regex=True, inplace = True)
    exploded.dropna(subset=[target_col], inplace=True)
    if new_col == None:
        return(pd.DataFrame(exploded[[target_col,*argv]]))
    else:
        exploded[new_col] = exploded[target_col]
        return(pd.DataFrame(exploded[[new_col,*argv]]))

if __name__ == '__main__':
    args = arg_parse()
    print(splitter(df, **args))

, затем вы выполняете этот код, вызывая это

python sep.py -t CountryRefs -s ','

или так

python sep.py -t CountryRefs -s ',' Friends Colors

...