Я изучаю коалы и пытаюсь преобразовать какой-то существующий код pandas. Мы используем один общий шаблон column.map(lambda x: x...). Иногда это работает без каких-либо дополнительных модификаций, а иногда нет.
Как я могу сделать это, если тип возвращаемого значения list ?. Кажется, исходный код указывает, что любой тип python должен быть приемлемым в качестве подсказки типа для @pandas_wraps, но я получаю ошибку ValueError: Missing type information. Ниже приведен минимальный пример - (я также пытался использовать str и object в качестве подсказки типа с одинаковыми результатами)
from databricks.koalas import pandas_wraps
data = {'col': ['test1\ntest2', 'test3\ntest4\ntest5']}
df = ks.DataFrame(data=data)
@pandas_wraps(return_col=list)
def split_col(x):
return x.split("\n")
df['col_split'] = df.col.map(split_col)
Полное сообщение об ошибке:
ValueError: Missing type information. It should either be provided as an argument to pandas_wraps, or as a python typing hint