Question

我想根据系列中的整数切片数据框（只包含字符串）中的列。这是一个例子：

data = pandas.DataFrame(['abc','scb','dvb'])
indices = pandas.Series([0,1,0])

然后应用一些功能，我得到以下内容：

Answer 1

您可以使用以下矢量化方法：

In [191]: [tuple(x) for x in indices.reset_index().values]
Out[191]: [(0, 0), (1, 1), (2, 0)]

In [192]: data[0].str.extractall(r'(.)') \
                 .loc[[tuple(x) for x in indices.reset_index().values]]
Out[192]:
         0
  match
0 0      a
1 1      c
2 0      d

In [193]: data[0].str.extractall(r'(.)') \
                 .loc[[tuple(x) for x in indices.reset_index().values]] \
                 .reset_index(level=1, drop=True)
Out[193]:
   0
0  a
1  c
2  d

说明：

In [194]: data[0].str.extractall(r'(.)')
Out[194]:
         0
  match
0 0      a
  1      b
  2      c
1 0      s
  1      c
  2      b
2 0      d
  1      v
  2      b

In [195]: data[0].str.extractall(r'(.)').loc[ [ (0,0), (1,1) ] ]
Out[195]:
         0
  match
0 0      a
1 1      c

Numpy解决方案：

In [259]: a = np.array([list(x) for x in data.values.reshape(1, len(data))[0]])

In [260]: a
Out[260]:
array([['a', 'b', 'c'],
       ['s', 'c', 'b'],
       ['d', 'v', 'b']],
      dtype='<U1')

In [263]: pd.Series(a[np.arange(len(data)), indices])
Out[263]:
0    a
1    c
2    d
dtype: object

Answer 2

您可以使用python预先操作列表。

l1 = ['abc','scb','dvb']
l2 = [0,1,0]
l3 = [l1[i][l2[i]] for i in range(len(l1))]

你得到l3为

['a', 'c', 'd']

现在将其转换为DataFrame

data = pd.DataFrame(l3)

您可以获得所需的数据框

基于整数数组在python pandas DataFrame中切割字符串

2 个答案: