使用冗长的列名称,无论设置options是什么,DataFrame都会以非常混乱的形式显示。
信息:我在Jupyter QtConsole,pandas 0.20.1,在启动时指定了以下相关选项:
pd.set_option('display.max_colwidth', 20)
pd.set_option('expand_frame_repr', False)
pd.set_option('display.max_rows', 25)
问题:如何在必要时截断DataFrame而不是将列包装到下一行,同时保留expand_frame_repr=False
?
这是一个例子。同样,问题不依赖于列数而是列的长度。
这不会导致问题:
df = pd.DataFrame(np.random.randn(1000, 1000),
columns=['col' + str(i) for i in range(1000)])
具有长列名的相同DataFrame会导致我正在讨论的问题:
df = pd.DataFrame(np.random.randn(1000, 1000),
columns=['very_long_col_name_'
+ str(i) for i in range(1000)])
有没有办法让第二个输出符合我失踪的第一个输出? (通过指定选项,而不是每次我想查看时都使用.iloc
。)
答案 0 :(得分:2)
使用max_columns
from string import ascii_letters
df = pd.DataFrame(np.random.randint(10, size=(5, 52)), columns=list(ascii_letters))
with pd.option_context(
'display.max_colwidth', 20,
'expand_frame_repr', False,
'display.max_rows', 25,
'display.max_columns', 5,
):
print(df.add_prefix('really_long_column_name_'))
really_long_column_name_a really_long_column_name_b ... really_long_column_name_Y really_long_column_name_Z
0 8 1 ... 1 9
1 8 5 ... 2 1
2 5 0 ... 9 9
3 6 8 ... 0 9
4 1 2 ... 7 1
[5 rows x 52 columns]
另一个想法......显然不完全是你想要的,但也许你可以根据自己的需要扭转它。
d1 = df.add_suffix('_really_long_column_name')
with pd.option_context('display.max_colwidth', 4, 'expand_frame_repr', False):
mw = pd.get_option('display.max_colwidth')
print(d1.rename(columns=lambda x: x[:mw-3] + '...' if len(x) > mw else x))
a... b... c... d... e... f... g... h... i... j... ... Q... R... S... T... U... V... W... X... Y... Z...
0 6 5 5 5 8 3 5 0 7 6 ... 9 0 6 9 6 8 4 0 6 7
1 0 5 4 7 2 5 4 3 8 7 ... 8 1 5 3 5 9 4 5 5 3
2 7 2 1 6 5 1 0 1 3 1 ... 6 7 0 9 9 5 2 8 2 2
3 1 8 7 1 4 5 5 8 8 3 ... 3 6 5 7 1 0 8 1 4 0
4 7 5 6 2 4 9 7 9 0 5 ... 6 8 1 6 3 5 4 2 3 2
答案 1 :(得分:2)
看起来需要增强功能。 repr
函数中的相关代码似乎在此处:
max_rows = get_option("display.max_rows")
max_cols = get_option("display.max_columns")
show_dimensions = get_option("display.show_dimensions")
if get_option("display.expand_frame_repr"):
width, _ = console.get_console_size()
else:
width = None
self.to_string(buf=buf, max_rows=max_rows, max_cols=max_cols,
line_width=width, show_dimensions=show_dimensions)
因此,要么传递expand_frame_repr=True
并且它包裹线宽,要么传递expand_frame_repr=False
,它不应该传递。但看起来代码中有一个错误(这应该是pandas 0.20.3 iirc):
pd.io.formats.format.DataFrameFormatter
中的:
def _chk_truncate(self):
"""
Checks whether the frame should be truncated. If so, slices
the frame up.
"""
from pandas.core.reshape.concat import concat
# Column of which first element is used to determine width of a dot col
self.tr_size_col = -1
# Cut the data to the information actually printed
max_cols = self.max_cols
max_rows = self.max_rows
if max_cols == 0 or max_rows == 0: # assume we are in the terminal
# (why else = 0)
(w, h) = get_terminal_size()
self.w = w
self.h = h
if self.max_rows == 0:
dot_row = 1
prompt_row = 1
if self.show_dimensions:
show_dimension_rows = 3
n_add_rows = (self.header + dot_row + show_dimension_rows +
prompt_row)
# rows available to fill with actual data
max_rows_adj = self.h - n_add_rows
self.max_rows_adj = max_rows_adj
# Format only rows and columns that could potentially fit the
# screen
if max_cols == 0 and len(self.frame.columns) > w:
max_cols = w
if max_rows == 0 and len(self.frame) > h:
max_rows = h
看起来它打算做你想做的事,但未完成。它会根据列的数字检查max_cols
,而不是列的总宽度。
因此,您可以创建一个show_df
函数来计算正确的列数并将其显示在option_context
中,如pi2Squared的答案,或者在此处修复(如果需要,可以提交补丁)它分发了。)
答案 2 :(得分:0)
正如其他人所指出的那样,熊猫本身似乎在这里被窃听或设计错误,因此需要一种解决方法。
大多数情况下,数字列会出现此问题,因为数字相对较短。如果列中有空格,Pandas会将列标题拆分为多行,因此当您显示数据帧时,可以通过在数字列的列标题中插入空格来“入侵”正确的行为。我有一个单行代码来执行此操作:
def colfix(df, L=5): return df.rename(columns=lambda x: ' '.join(x.replace('_', ' ')[i:i+L] for i in range(0,len(x),L)) if df[x].dtype in ['float64','int64'] else x )
显示您的数据框,只需输入
即可colfix(your_df)
请注意,重命名不会永久更改数据框,它只会为名称添加空格,以便一次显示它。
结果(在Jupyter笔记本中):
使用colfix
:
不