Question

我有一个数据帧x：

x = pd.DataFrame(np.random.randn(3,3), index=[1,2,3], columns=['A', 'B', 'C'])
x


       A    B   C
1   0.256668    -0.338741   0.733561
2   0.200978    0.145738    -0.409657
3   -0.891879   0.039337    0.400449

我想选择一堆索引列对来填充新系列。例如，我可以选择[（1，A），（1，B），（1，A），（3，C）]，它们将生成一个包含4个元素的列表或数组或系列：

[0.256668, -0.338741, 0.256668, 0.400449]

知道我应该怎么做吗？

Answer 1

我认为get_value()和lookup()更快：

import numpy as np
import pandas as pd
x = pd.DataFrame(np.random.randn(3,3), index=[1,2,3], columns=['A', 'B', 'C'])

locations = [(1, "A"), (1, "B"), (1, "A"), (3, "C")]

print x.get_value(1, "A")

row_labels, col_labels = zip(*locations)
print x.lookup(row_labels, col_labels)

Answer 2

使用 ix 应该能够找到数据框中的元素，如下所示：

import pandas as pd

# using your data sample
df = pd.read_clipboard()

df
Out[170]: 
          A         B         C
1  0.256668 -0.338741  0.733561
2  0.200978  0.145738 -0.409657
3 -0.891879  0.039337  0.400449

# however you cannot store A, B, C... as they are undefined names
l = [(1, 'A'), (1, 'B'), (1, 'A'), (3, 'C')]

# you can also use a for/loop, simply iterate the list and LOCATE the element
map(lambda x: df.ix[x[0], x[1]], l)
Out[172]: [0.25666800000000001, -0.33874099999999996, 0.25666800000000001, 0.400449]

Answer 3

如果您的配对是位置而不是索引/列名，

row_position = [0,0,0,2]
col_position = [0,1,0,2]

x.values[row_position, col_position]

或者从np.searchsorted

获取职位

row_position = np.searchsorted(x.index,row_labels,sorter = np.argsort(x.index))

从pandas dataframe中选择特定的索引，列对

3 个答案: