def execute_series_translate_series_series(
op, data, from_string, to_string, **kwargs
):
to_string_iter = iter(to_string)
table = from_string.apply(
lambda x, y: maketrans(x, y=next(y)), args=(to_string_iter,)
)
return data.str.translate(table)
def _ibis_sqlite_translate(string, from_string, to_string):
if (string is not None and from_string is not None
and to_string is not None):
table = maketrans(from_string, to_string)
return string.translate(table)
return None
def execute_series_translate_scalar_scalar(
op, data, from_string, to_string, **kwargs
):
return data.str.translate(maketrans(from_string, to_string))
def execute_series_translate_scalar_series(
op, data, from_string, to_string, **kwargs
):
table = to_string.map(lambda y, x=from_string: maketrans(x=x, y=y))
return data.str.translate(table)
id='re_extract',
),
param(
lambda t: t.string_col.re_replace(r'[[:digit:]]+', 'a'),
lambda t: t.string_col.str.replace(r'\d+', 'a'),
id='re_replace',
),
param(
lambda t: t.string_col.repeat(2),
lambda t: t.string_col * 2,
id='repeat'
),
param(
lambda t: t.string_col.translate('0', 'a'),
lambda t: t.string_col.map(decode).str.translate(
maketrans(u'0', u'a')
),
id='translate',
),
param(
lambda t: t.string_col.find('a'),
lambda t: t.string_col.str.find('a'),
id='find'
),
param(
lambda t: t.string_col.lpad(10, 'a'),
lambda t: t.string_col.str.pad(10, fillchar='a', side='left'),
id='lpad'
),
param(
lambda t: t.string_col.rpad(10, 'a'),
param(
lambda t: t.string_col.re_extract(r'([[:digit:]]+)', 0),
lambda t: t.string_col.str.extract(r'(\d+)', expand=False),
id='re_extract',
),
param(
lambda t: t.string_col.re_replace(r'[[:digit:]]+', 'a'),
lambda t: t.string_col.str.replace(r'\d+', 'a'),
id='re_replace',
),
param(lambda t: t.string_col.repeat(2),
lambda t: t.string_col * 2,
id='repeat'),
param(
lambda t: t.string_col.translate(u'0', u'a'),
lambda t: t.string_col.str.translate(maketrans(u'0', u'a')),
id='translate',
),
param(lambda t: t.string_col.find('a'),
lambda t: t.string_col.str.find('a'),
id='find'),
param(lambda t: t.string_col.lpad(10, 'a'),
lambda t: t.string_col.str.pad(10, fillchar='a', side='left'),
id='lpad'),
param(
lambda t: t.string_col.rpad(10, 'a'),
lambda t: t.string_col.str.pad(10, fillchar='a', side='right'),
id='rpad',
),
param(
lambda t: t.string_col.find_in_set(['1']),
