有没有一只Pandas和特里布尔相当？

Question

我想为python编写单元测试，声明一些简单的数据帧。 有没有一只熊猫相当于Rtribble。例如：

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

你需要DataFrame

pd.DataFrame({'colA':['a','b','c'],'col2':[1,2,3]})
Out[391]: 
  colA  col2
0    a     1
1    b     2
2    c     3

Answer 2

您可以定义自己的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
)