使用stringstream、setw和setprecision格式化矩阵的问题

问题描述 #

我尝试使用stringstream、setw和setprecision格式化矩阵，代码如下：

template <typename T>
std::string format(std::vector<T> &a, int m, int n, int lda)
{
    std::stringstream ss;
    for (int i = 0; i < m; i++)
    {
        for (int j = 0; j < n; j++)
        {
            ss << std::setw(6);
            ss << a[i + j * lda];
            ss << std::setprecision(2);
        }
        if (i < m - 1)
            ss << std::endl;
    }
    return ss.str();
}

int main()
{
    const int LDA = 5;
    const int N = LDA;

    std::vector<double> a = {
        6.39, 0.00, 0.00, 0.00, 0.00,
        0.13, 8.37, 0.00, 0.00, 0.00,
        -8.23, -4.46, -9.58, 0.00, 0.00,
        5.71, -6.10, -9.25, 3.72, 0.00,
        -3.18, 7.21, -7.42, 8.54, 2.51};

    std::cout << format(a, N, N, LDA) << std::endl;

    return 0;
}

实际输出 #

6.39  0.13  -8.2   5.7  -3.2
     0   8.4  -4.5  -6.1   7.2
     0     0  -9.6  -9.2  -7.4
     0     0     0   3.7   8.5
     0     0     0     0   2.5

疑问 #

• 为何-8.2、8.4这类数值会被舍入？
• 为何零未被格式化？

期望输出 #

6.39   0.13  -8.23   5.71  -3.18
   0.00   8.37  -4.46  -6.10   7.21
   0.00   0.00  -9.58  -9.25  -7.42
   0.00   0.00   0.00   3.72   8.54
   0.00   0.00   0.00   0.00   2.51

解决方案 #

问题原因分析 #

1. 数值舍入问题：默认情况下setprecision(n)设置的是有效数字个数，而非小数位数。你把setprecision(2)放在第一次输出之后，导致从第二个数值开始，所有数字都被限制为2位有效数字，比如-8.23会被截断为-8.2，8.37会四舍五入为8.4。
2. 零未格式化问题：同样因为有效数字规则，0.00的2位有效数字就是0，且默认浮点数输出会省略末尾的零和无意义的小数点，所以只显示0。

修正措施 #

要实现期望的格式，需要设置三个关键格式标志：

• std::fixed：强制以固定小数位数输出，此时setprecision(n)表示保留n位小数。
• std::showpoint：强制显示小数点和末尾的零，确保所有数值都以X.XX的形式呈现。
• 调整setw宽度：每个数值需要容纳符号（如负号）、整数部分、小数点、两位小数，设置setw(7)可以保证对齐（例如"-8.23"占5个字符，加上两侧空格后总宽度7，能和其他数值对齐）。
• 格式标志只需设置一次：fixed、showpoint、setprecision是持久生效的，不需要在每个循环里重复设置；只有setw是单次生效，需要每次输出前设置。

修正后的代码 #

#include <iomanip> // 必须包含这个头文件以使用格式控制符

template <typename T>
std::string format(std::vector<T> &a, int m, int n, int lda)
{
    std::stringstream ss;
    // 设置持久生效的格式标志
    ss << std::fixed << std::showpoint << std::setprecision(2);
    for (int i = 0; i < m; i++)
    {
        for (int j = 0; j < n; j++)
        {
            ss << std::setw(7); // 每次输出前设置宽度
            ss << a[i + j * lda];
        }
        if (i < m - 1)
            ss << std::endl;
    }
    return ss.str();
}

// main函数保持不变
int main()
{
    const int LDA = 5;
    const int N = LDA;

    std::vector<double> a = {
        6.39, 0.00, 0.00, 0.00, 0.00,
        0.13, 8.37, 0.00, 0.00, 0.00,
        -8.23, -4.46, -9.58, 0.00, 0.00,
        5.71, -6.10, -9.25, 3.72, 0.00,
        -3.18, 7.21, -7.42, 8.54, 2.51};

    std::cout << format(a, N, N, LDA) << std::endl;

    return 0;
}

验证输出 #

运行修正后的代码，输出结果与期望完全一致：

6.39   0.13  -8.23   5.71  -3.18
   0.00   8.37  -4.46  -6.10   7.21
   0.00   0.00  -9.58  -9.25  -7.42
   0.00   0.00   0.00   3.72   8.54
   0.00   0.00   0.00   0.00   2.51

内容的提问来源于stack exchange，提问作者ramgorur