☰

Диаграмма рассеяния с данными из CSV-файла и линией тренда

tag-icon • tag-icon tikz-pgf pgfplots plot csv
Диаграмма рассеяния с данными из CSV-файла и линией тренда

Я создал диаграмму рассеивания с помощью кода изэтот ранее заданный вопрос.

\documentclass[varwidth=true, border=2pt]{standalone}
\usepackage[utf8]{inputenc} % this is needed for umlauts
\usepackage[ngerman]{babel} % this is needed for umlauts
\usepackage[T1]{fontenc}    % this is needed for correct output of umlauts in pdf
\usepackage[margin=2.5cm]{geometry} %layout  
\usepackage{pgfplots}


\begin{filecontents}{table3.csv}
column 1    column 2
1966    37.51817228
1960    40.56693583
1961    40.71972964
1962    40.97560208
1964    41.11687187
1963    41.25082828
1965    46.02625404
1960    46.22815872
1967    46.67800113
1961    48.39523271
\end{filecontents}


\begin{document}

\begin{tikzpicture}
    \begin{axis}[
            axis x line=middle,
            axis y line=middle,
            enlarge y limits=true,
            %xmin=0, xmax=2150,
            %ymin=0, ymax=600,
            width=15cm, height=8cm,     % size of the image
            grid = major,
            grid style={dashed, gray!30},
            ylabel=steps,
            xlabel=$n$,
            legend style={at={(0.1,-0.1)}, anchor=north}
         ]        
          \addplot[scatter,only marks] table [x=column 1, y=column 2, col sep=comma] {table3.csv};

          %the code below is added via @Peter's comment.
          \addplot[only marks] table [col sep = comma,y={create col/linear regression={y=column 2}}]{table3.csv};


    \end{axis}
\end{tikzpicture}

\end{document}

Диаграмма рассеивания в итоге получается отличной, но я хочу иметь возможность добавить линию тренда. Все примеры линий тренда, которые я видел, были рассчитаны с данными, введенными непосредственно в файл .TeX, а не из файла .csv.

Можно ли сделать это?

Другая моя мысль была продолжить в Excel, вычислить линию тренда, а затем наложить линию на график. Я бы предпочел сделать это более прямым способом, так как в моем документе много графиков.

редактировать:Джейк дал мне отличное руководство о том, как это сделать с данными, напрямую введенными в файл TeX, но у меня возникли проблемы с парсингом напрямую из файла .csv. Я добавил в свой код и опубликовал сообщение об ошибке, которое получаю в консоли.

После добавления кода я получаю сообщение об ошибке.

./linearreg.tex:30: Package PGF Math Error: Could not parse input '' as a float
ing point number, sorry. The unreadable part was near ''..

Строка 30 в моем документе — это добавленная строка с уравнением линейной регрессии.

окончательная правка:Я разобрался. Эта ошибка была вызвана тем, что в моем файле были столбцы с пустыми данными. Мне пришлось удалить эти пустые данные, чтобы вычислить линию линейной регрессии.

Вот окончательный вариант моего кода:

\documentclass{article}
\usepackage{tikz}
\usepackage{pgfplots}
\usepackage{pgfplotstable}

\begin{document}
\pgfplotstableread[col sep = comma]{table4.csv}\loadedtable

\begin{tikzpicture}
    \begin{axis}[
        xlabel=Weight (kg), % label x axis
        ylabel=Height (cm), % label y axis
        axis lines=left, %set the position of the axes
        clip=false
    ]

            \addplot[scatter, only marks] table [x=column 1, y=column 2, col sep=comma] {\loadedtable};
            \addplot[very thick, red] table [col sep = comma,y={create col/linear regression={y=column 2}}]{\loadedtable};

    \end{axis}

\end{tikzpicture}
\end{document}

решение1

Чтобы получить линию линейной регрессии для данных из файла данных, используйте

\addplot [no markers] table [y={create col/linear regression={y=<column name>}}] {<file name>};


\documentclass[border=5mm]{standalone}
\usepackage{pgfplots, pgfplotstable}
\usepackage{filecontents}

\begin{filecontents}{table.dat}
x y
0 1
100 3
150 2
200 5
300 6
\end{filecontents}

\begin{document}

\begin{tikzpicture}
    \begin{axis}[
            axis x line=middle,
            axis y line=middle,
            enlarge y limits=true,
            width=15cm, height=8cm,     % size of the image
            grid = major,
            grid style={dashed, gray!30},
            ylabel=steps,
            xlabel=$n$,
            legend style={at={(0.1,-0.1)}, anchor=north}
         ]        
        \addplot[only marks] table  {table.dat};
        \addplot [no markers, thick, red] table [y={create col/linear regression={y=y}}] {table.dat};
    \end{axis}
\end{tikzpicture}

\end{document}

Связанный контент