Início / coding / Perguntas / 78749249
Accepted
ignoring_gravity
Asked: 2024-07-15 18:05:17 +0800 CST

Como anexar string a cada elemento da matriz em partes?

  • 772

Diga que eu tenho

In [22]: import pyarrow as pa

In [23]: t = pa.table({'a': ['one', 'two', 'three']})

e gostaria de acrescentar '_frobenius'a cada elemento de'a'

Resultado esperado:

pyarrow.Table
a: string
----
a: [["one_frobenius","two_frobenius","three_frobenius"]]
python

2 respostas

  1. Best Answer

    Você pode usar pyarrow.compute.binary_join_element_wisepara juntar uma string fixa a cada elemento:

    import pyarrow as pa
import pyarrow.compute as pc

t = pa.table({"a": ["one", "two", "three"]})

t = t.set_column(0, "a", pc.binary_join_element_wise(t["a"], "_frobenius", ""))

print(t)

    Saída:

    pyarrow.Table
a: string
----
a: [["one_frobenius","two_frobenius","three_frobenius"]]
    • 3

  2. Você pode converter para o Dataframe do pandas, modificar as strings e converter novamente. Por exemplo,

    >>> import pyarrow as pa
>>> t = pa.table({'a': ['one', 'two', 'three']})
>>> df = t.to_pandas()
>>> df
       a
0    one
1    two
2  three
>>> df['a'] = df['a'] + '_frobenius'
>>> df
                 a
0    one_frobenius
1    two_frobenius
2  three_frobenius
>>> new_table = t.set_column(0, 'a', pa.array(df['a']))
>>> new_table
pyarrow.Table
a: string
----
a: [["one_frobenius","two_frobenius","three_frobenius"]]
    • 0

relate perguntas