列表内数组中元素的总和

Question

I'm trying to calculate the np.sum of the elements within each array. I tried in stead of np.sum(outcome_list[0] == 'H' to just leave it as np.sum(outcome_list[j] == 'H' so that each "list" would have its own data set on the total number of heads, but it didn't like it. The bigger question is, how would I construct an array with a given base list and the action to be done in each element of that list?

EDIT:

the throw_a_coin definition

def throw_a_coin(N):
    return np.random.choice(['H','T'], size=N)
N =40

trials (as shown above) is the set to be acted upon

for i in trials:
    throws = throw_a_coin(i)
    outcome_list.append(throws)

for j in outcome_list:
    print("Number of Heads:", np.sum(outcome_list[0] == 'H'))
    print (j)

EDIT 2:

problem resolved with the one shown below, however I'm getting more than 13 numbers for "probabilities" - it seems that the system is running through the trials list more than once.

def throw_a_coin(N):
    return np.random.choice(['H','T'], size=N)

trials = [10, 30, 50, 70, 100, 130, 170, 200, 500, 1000, 2000, 5000, 10000]

for i in trials:
    throws = throw_a_coin(i)
    outcome_list.append(throws)

probabilities = []

for j in outcome_list:
    print("Number of Heads:", np.sum(j == 'H'))
    print("Number of Throws:", len(j))
    print("p = Number of Heads/Total Throws:", (np.sum(j == 'H'))/len(j))
    probabilities.append((np.sum(j =='H'))/len(j))
    print (j)
    print("\n")

print(probabilities)

Answer 1

尝试这个：

print("Number of Heads:", (j == 'H').sum())

Answer 2

You were nearly there! You just needed to replace

print("Number of Heads:", np.sum(outcome_list[0] == 'H'))

with

print("Number of Heads:", np.sum(j == 'H'))

Here is the complete answer:

trials = [10, 30, 50, 70, 100, 130, 170, 200, 500, 1000, 2000, 5000, 10000]

N =40
def throw_a_coin(N):
    return np.random.choice(['H','T'], size=N)

outcome_list = []
for i in trials:
    throws = throw_a_coin(i)
    outcome_list.append(throws)

for j in outcome_list:
    print("Number of Heads:", np.sum(j == 'H'))
    print (j)

Answer 3

如果output_list是一个numpy数组，并且矩阵中的每一行代表每个试验，则可以按如下所示的有效方式获得总和：

head_sum = output_list.sum(axis=1)

Answer 4

I would use a list comprehension followed by a len call.

for j in outcome_list:
    print("Number of Heads:", len([x for x in j if x == 'H'])
    print (j)

DISCLAIMER : I have zero experience with Numpy. However, this would be the general Pythonic way to do it without resorting to count .