[python字符串的索引与切片操作_python – Pandas切片字符串与另一列的索引

我想使用来自另一列的索引来切割此字符串.我得到NaN而不是字符串的片段.

import pandas as pd

from pandas import DataFrame, Series

sales = {'name': ['MSFTCA', 'GTX', 'MSFTUSA', ],

'n_chars': [2, 2, 3],

'Jan': [150, 200, 50],

'Feb': [200, 210, 90],

'Mar': [140, 215, 95]}

df = pd.DataFrame.from_dict(sales)

df

def extract_location(name, n_chars):

return( name.str[-n_chars:])

df.assign(location=(lambda x: extract_location(x['name'], x['n_chars']))).to_dict()

得到：

{'Feb': {0: 200, 1: 210, 2: 90},

'Jan': {0: 150, 1: 200, 2: 50},

'Mar': {0: 140, 1: 215, 2: 95},

'location': {0: nan, 1: nan, 2: nan},

'n_chars': {0: 2, 1: 2, 2: 3},

'name': {0: 'MSFTCA', 1: 'GTX', 2: 'MSFTUSA'}}