ykp.data
Out[182]:
state action reward
0 [41] 5 59
1 [5] 52 48
2 [46] 35 59
3 [42] 16 12
4 [43] 37 48
5 [36] 5 59
6 [49] 52 48
7 [39] 11 23
我想在状态条目中找到与 [42] 匹配的行,所以我跑了
ykp.data.query('state == [42]')
但我得到
Empty DataFrame
Columns: [state, action, reward]
Index: []
当我应该看到的时候[42], 16, 12
.
有人可以告诉我如何解决这个问题吗?我需要将状态值存储为数组。
最好避免pd.Series.apply
这里。相反,您可以使用itertools.chain
构造一个常规的 NumPy 数组。然后将数组与整数进行比较,形成用于索引的布尔数组:
from itertools import chain
df = pd.DataFrame(np.random.randint(0, 100, size=(100000, 1)), columns=['state'])
df = df.assign(state=df.state.apply(lambda x: [x]), axis=1)
def wen(df):
df.state=df.state.astype(str)
return df.query("state == '[42]'")
%timeit df[np.array(list(chain.from_iterable(df['state'].values))) == 42] # 14.2 ms
%timeit df[df.state.apply(tuple) == (42,)] # 41.9 ms
%timeit df.loc[df.state.apply(lambda x: x==[42])] # 33.9 ms
%timeit wen(df) # 19.9 ms
更好的是,不要在数据框中使用列表。只需使用常规的int
系列。这将提高内存和性能效率。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)