python dataframe 随机找出某一行中列值最大的位置[列值最大可能会有多个，随机挑选]

最新推荐文章于 2023-03-25 12:11:39 发布

wang603603

最新推荐文章于 2023-03-25 12:11:39 发布

阅读量4.8k

点赞数

分类专栏： python 文章标签： pthon dataframe

本文链接：https://blog.csdn.net/wang603603/article/details/82852422

版权

python 专栏收录该内容

17 篇文章 4 订阅

订阅专栏

# 找出 q_table 中第二行, 列值最大的位置.[答案为 a2 或者 a3]
q_table 
    a1  a2  a3   a4
0   1  11  11  111
1   2  22  22   22
2   3  33  33    1

#------------------------
q_table = pd.DataFrame({'a1': [1, 2, 3], 'a2': [11, 22, 33], 'a3': [11, 22, 33], 'a4': [111, 22, 1] })
print("q_table \n", q_table)


print(" ------------ ")
state_action = q_table.loc[2, :]
print(" origin \n",state_action)
state_action = state_action.reindex(np.random.permutation(state_action.index))
print(" reindex \n", state_action)
print(" ------------ ")
action = state_action.idxmax()
print(" max value's   index",action)

# 随机访问
action_index = np.random.choice(len(state_action))
action = state_action.index[action_index]