首页 > 其他分享 >用函数创建DataFrame:使用字典推导式

用函数创建DataFrame:使用字典推导式

时间:2024-10-08 16:48:48浏览次数:15  
标签:推导 df cols DataFrame 索引 ind 字典


import pandas as pd

def make_df(cols, ind):
    """一个简单的DataFrame"""
    # 字典推导式,为每列生成数据
    data = {c: [str(c) + str(i) for i in ind] for c in cols}
    # 创建并返回 DataFrame
    return pd.DataFrame(data, ind)

# DataFrame 示例
df = make_df('ABC', range(3))
print(df)

这段代码定义了一个名为 make_df 的函数,它接受两个参数:cols 和 ind。cols 是一个字符串,表示 DataFrame 的列名,ind 是一个可迭代对象,表示 DataFrame 的索引。

函数的主体是一个字典推导式,它为每列生成一个数据列表。每个元素是列名和索引值的组合。然后,它使用这个字典和提供的索引来创建并返回一个 DataFrame。

在这个示例中,cols 参数是 'ABC',它包含三个列名(A、B、C)。ind 参数是 range(3),它生成一个从 0 到 2 的索引列表。

运行这段代码后,你会得到以下输出:

   A   B   C
0  A0  B0  C0
1  A1  B1  C1
2  A2  B2  C2

每一列的数据是由列名和索引值组合而成的字符串。例如,在第一行,A 列的值是 'A0',B 列的值是 'B0',C 列的值是 'C0'。

标签:推导,df,cols,DataFrame,索引,ind,字典
From: https://www.cnblogs.com/redufa/p/18452008

相关文章

  • 洛谷每日一题(P2580 于是他错误的点名开始了)字典树/哈希表
    原题目链接:P2580于是他错误的点名开始了-洛谷|计算机科学教育新生态(luogu.com.cn)原题目截图:思路分析:解法一:哈希表法显而易见的一种思路,我们不妨模拟一下:当教练每次点名,我作为特派员,便查看一下有没有这个学生,是不是点过了这个学生。我们查看的过程,就依赖于一张表......
  • 洛谷每日一题(P1481 魔族密码)字典树解法
    原题目链接:P1481魔族密码-洛谷|计算机科学教育新生态(luogu.com.cn)原题目截图:思路分析:这道题的话其实有很多种方法,可以用动态规划做,不过我一看到这道题,脑子里不禁蹦出一个数据结构:“字典树”!字典树+深度优先搜索。那么在这之前,我们先来了解一下什么是字典树吧!什......