python - 在 pandas 数据帧字符串列中第 n 行换行后丢弃字符串

Question

我有一个包含几列的 pandas 数据框。看其中一个，输入 string，我的目标是丢弃第 500 个换行符之后的每个字符。

到目前为止我的方法是:循环n=len(string)、计数\n 和split()。我可以做一些更优雅、更高效的事情吗？

Answer 1

当您说计算“\n”时，我不确定您使用了什么，但使用 join() 和 split() 函数是非常干净的方法:

a = "a \n b \n c \n d"

"".join(a.split("\n")[0:2])

'a  b '

由于您正在使用数据框，因此您可以将其与列表理解结合起来以获得简洁的结果:

df = pd.DataFrame({"A" : ["a \n b \n c \n d", "d \n e \n f \n g"]})

df['A'] = ["".join(x.split("\n")[0:2]) for x in df['A']]

     A
0  a  b 
1  d  e