Question

您好我想找到整数数组的最大值和最小值的第一个索引。

我的代码在重复值的情况下返回所有索引...

tf.reset_default_graph()
    saver = tf.train.import_meta_graph(trained_model_name)

    with tf.Session() as sess:
        saver.restore(sess, tf.train.latest_checkpoint('./'))

    graph = tf.get_default_graph()

    X_init = tf.placeholder(tf.float32, shape=(c.vocab_size, c.emb_size))

    input_data = graph.get_tensor_by_name("input_data")

    preds = graph.get_tensor_by_name("preds")

    init = tf.global_variables_initializer()

    sess.run(init, feed_dict={X_init: lexvec_model})

    pred = sess.run(preds, feed_dict={input_data: model_input})

输出： 0 1 五 7

我想要的：（0,5）

感谢您的帮助。

Answer 1

def minmaxloc(num_list):
    return num_list.index(max(num_list)), num_list.index(min(num_list))

print(minmaxloc(a))

Answer 2

使用numpy的argmin和argmax方法：

import numpy as np
def minmaxloc(num_list):
    return np.argmin(num_list), np.argmax(num_list)
A= [1,1,8,7,5,9,6,9]
print(minmaxloc(A))

如何查找int数组Python的最小值和最大值索引

2 个答案: