将列中的值用作另一列pandas的字符串切片机

Question

我正在尝试将单元格值用作新列中字符串的切片。例如，如果我创建此表。

data = pd.DataFrame(data = {'Name':['This is a title'], 'Number':[-5]})

               Name Number
0   This is a title     -5

并创建一个类似的新列：

data['Test'] = data.Name.str[:data.Number.item()]

它将创建新的列，如预期：

               Name Number       Test
0   This is a title     -5  This is a 

当我的行超过行时发生问题，因此，如果我创建下表：

 data = pd.DataFrame(data = {'Name':['This is a title', 'This is another title'], 'Number':[-5, -13]})

                     Name   Number
0   This is a title             -5
1   This is another title      -13

“测试”列的创建产生：

can only convert an array of size 1 to a Python scalar

我理解为什么这是因为该列现在具有多个值以来，我想知道的是如何使用具有多个行以上的数据框架来执行此操作？我尝试了.items（），.values（）等。新列只是NAN。

有什么想法吗？

谢谢！

Answer 1

您可以使用使用与axis = 1使用，然后按dataframe行进。

import pandas as pd
data = pd.DataFrame(data = {'Name':['This is a title', 'This is another title'], 'Number':[-5, -13]})

data['Test'] = data.apply(lambda row: row['Name'][:row['Number']], axis=1)
print(data)

输出：

                    Name  Number        Test
0        This is a title      -5  This is a 
1  This is another title     -13    This is 

Answer 2

不幸的是，在这里，您需要循环。列表理解将是最有效的：

data['Test'] = [s[:i] for s,i in zip(data['Name'], data['Number'])]

输出：

                    Name  Number        Test
0        This is a title      -5  This is a 
1  This is another title     -13    This is