往dataframe中插入大量的特征列

本文来自百度问心一言

已有一个dataframe df， 还有一个特征列表indis = ['fakebuy0', 'buy01', 'buy01', 'buy01', 'sell01', 'sell01'] , 如何以indis中的元素为列名，往df中插入所有列

当然，除了之前提到的方法，还有其他几种方式可以向 DataFrame 中添加新列。以下是一些额外的方法：

方法 1: 直接赋值

你可以直接通过列名给 DataFrame 添加新列。如果列名已经存在，它将被覆盖；如果不存在，Pandas 将会创建这个新列。

python复制代码

	for col in set(indis):
	df[col] = None # 或者使用其他默认值，如 0, np.nan, '' 等

方法 2: 使用 reindex

虽然 reindex 主要用于行索引，但也可以用于列。你可以构造一个新的列索引，然后使用 reindex 方法来扩展 DataFrame。

python复制代码

	new_columns = df.columns.tolist() + list(set(indis))
	df = df.reindex(columns=new_columns)

请注意，reindex 会引入缺失值（NaN），因为新添加的列在原始 DataFrame 中不存在。

方法 3: 使用 concat 或 merge

虽然这些方法通常用于合并两个 DataFrame，但你也可以创建一个新的 DataFrame，只包含新列，然后使用 concat 或 merge 将其与原始 DataFrame 合并。这种方法可能更复杂，但在某些情况下可能很有用，特别是当你需要合并来自不同数据源的数据时。

python复制代码

	# 创建一个只有新列的空 DataFrame
	new_df = pd.DataFrame(columns=list(set(indis)))
	# 使用 concat 合并，注意这里需要处理索引对齐问题
	df = pd.concat([df, new_df], axis=1)

方法 4: 使用 assign 与字典推导式

之前已经提到了 assign 方法，但你可以结合字典推导式来动态创建列并赋值。

python复制代码

df = df.assign(**{col: np.nan for col in set(indis)})

方法 5: 使用 insert 方法

Pandas 的 DataFrame 也有一个 insert 方法，它可以在指定位置插入一列。

python复制代码

	for col in set(indis):
	if col not in df.columns:
	df.insert(len(df.columns), col, np.nan) # 在最后一列插入新列

每种方法都有其用途，选择哪种方法取决于你的具体需求和偏好。对于简单的列添加，直接赋值或使用 assign 方法通常是最直接和最简单的。如果你需要更复杂的操作，比如合并来自不同源的数据或在特定位置插入列，那么 concat、merge 或 insert 可能更合适。

reindex 可以指定默认值吗？

是的，reindex 方法可以指定默认值，这在为新引入的索引填充缺失值时非常有用。在 Pandas 中，reindex 方法有一个参数叫做 fill_value，你可以通过这个参数为新添加的索引位置指定一个默认值。

下面是一个简单的例子，展示了如何使用 reindex 方法并指定默认值：

python复制代码

	import pandas as pd
	# 创建一个简单的 DataFrame
	df = pd.DataFrame({
	'A': [1, 2, 3],
	'B': [4, 5, 6]
	})
	# 原始的 DataFrame
	print("Original DataFrame:")
	print(df)
	# 假设我们想要添加新列，并使用 reindex 方法
	new_columns = df.columns.tolist() + ['new_column1', 'new_column2']
	# 使用 reindex 添加新列，并为新列指定默认值为 0
	df_reindexed = df.reindex(columns=new_columns, fill_value=0)
	# 输出结果
	print("\nDataFrame after reindexing:")
	print(df_reindexed)

在这个例子中，reindex 方法被用于扩展 DataFrame 的列，通过 fill_value=0 参数，所有新添加的列都将被填充为 0。

请注意，reindex 不仅可以用于列，还可以用于行。在用于行时，你也可以指定 fill_value 来为新增的行索引填充默认值。