Question

我具有该功能：

def count (a,b):
    x = a*b

“ a”和“ b”的值必须为1 ... 99（对于“ a”）和100 ... 800（对于“ b”）。因此，问题是如何创建垂直方向为a值，水平为b值，内部为x值（使用a和b的所有组合）进行计数的熊猫数据框？它必须看起来像这样： example

Answer 1

这可以通过矩阵乘法轻松完成：

import pandas as pd
import numpy as np

a = np.arange(1,100)
b = np.arange(100,801)

df = pd.DataFrame(np.matmul(a.reshape(-1,1),b.reshape(1,-1), index=a, columns=b)

Answer 2

希望这会有所帮助

import pandas as pd

def count(a,b):
    x = a*b
    return x

a = list(range(1,100))
b = list(range(100,801))
data = []
for i in a:
    temp = [i]
    for j in b:
        temp.append(count(i,j))
    data.append(temp)

df = pd.DataFrame(data, columns=["a/b"]+b)
# to save as csv
df.to_csv("data.csv", index=False)

如何创建熊猫数据框并通过函数填充它？

2 个答案: