python - 有没有相当于 tribble 的 panda ？

Question

我想为 python 编写单元测试，声明一些简单的数据帧。有没有相当于R的 Pandas tribble 。例如:

tribble(
  ~colA, ~colB,
  "a",   1,
  "b",   2,
  "c",   3
)
#> # A tibble: 3 x 2
#>   colA   colB
#>   <chr> <dbl>
#> 1 a         1
#> 2 b         2
#> 3 c         3

Answer 1

您可以定义自己的tribble函数:

def tribble(columns, *data):
    return pd.DataFrame(
        data=list(zip(*[iter(data)]*len(columns))),
        columns=columns
    )

然后你可以像这样使用它:

tribble(
    ['colA', 'colB'],
    'a', 1,
    'b', 2,
    'c', 3
)