☰

Фильтрация строк таблицы для регрессии

tag-icon • tag-icon pgfplots pgfplotstable
Фильтрация строк таблицы для регрессии

У меня есть таблица, содержащая флаг и значения x и y. Я хотел бы получить регрессию для всех строк с 1 в первом столбце. Как это сделать?

\documentclass{article}
\usepackage{array}
\usepackage{pgfplots}
\pgfplotsset{compat=1.5.1}
\usepackage{pgfplotstable}
\begin{document}
  \pgfplotstableread{
    Exmpl  a   v
    1      0   0
    1      1   1
    1      2   1
    1      3   4
    2      0   -0
    2      1   -1
    2      2   -1
    2      3   -4
  } \data
  \begin{tikzpicture}
    \begin{axis} % [legend pos=outer north east]
      \addplot table [x=a, y=v] {\data};
      \addplot table [x=a, y={create col/linear regression={y=v}}] {\data};
    \end{axis}
  \end{tikzpicture}
\end{document}

Я думаю, фильтры pgfplots здесь не применимы, так как, как я предполагаю, pgfplots получает только значения x и y. Было бы здорово, если бы была возможность создать отфильтрованную таблицу и делать регрессию и построение графиков оттуда.

решение1

Думаю, у меня почти есть решение с помощью двух вопросов: Создайте таблицу сопряженности с помощью pgfplotstableи Доступ к отдельным элементам таблицы с помощью pgfplots? Идея состоит в том, чтобы транспонировать матрицу, создать новую матрицу, пройтись по всем столбцам и скопировать столбцы, если значение Exmpl равно 1.

\documentclass{article}
\usepackage{array}
\usepackage{pgfplots}
\pgfplotsset{compat=1.5.1}
\usepackage{pgfplotstable}
\usepackage{ifthen}

\newcommand{\pgfplotstablefilterrows}[3]
{
  \pgfplotstablegetrowsof{#1}
  \pgfmathsetmacro{\NumOfRows}{\pgfplotsretval}
  \pgfmathsetmacro{\MaxRow}{\NumOfRows-1}
  \pgfplotstablegetcolsof{#1}
  \pgfmathsetmacro{\NumOfCols}{\pgfplotsretval}

  \pgfplotstabletranspose{\TransposedData}{#1}
  \pgfplotstableset{create on use/TransposedHead/.style={copy column from table={\TransposedData}{[index]0}}}
  \pgfplotstablenew[columns={TransposedHead}]{\NumOfCols}{\TransposedFilteredData}
  \pgfplotsforeachungrouped \pgfplotstablerowindex in {0,1,...,\MaxRow}{ % Row loop
    #3
  }
  \pgfplotstabletranspose[colnames from=TransposedHead,input colnames to=]{#2}{\TransposedFilteredData}
  \pgfplotstableclear{\TransposedData}
  \pgfplotstableclear{\TransposedFilteredData}
}

\begin{document}
  \pgfplotstableread{
    Exmpl  a   v
    1      0   0
    1      1   1
    1      2   1
    1      3   4
    2      0   -0
    2      1   -1
    2      2   -1
    2      3   -4
  } \data

  \pgfplotstablefilterrows{\data}{\FilteredData}
  {
    \pgfplotstablegetelem{\pgfplotstablerowindex}{[index]0}\of\data
    \ifthenelse{\equal{\pgfplotsretval}{1}}
    {
      \pgfplotstablecreatecol[copy column from table={\TransposedData}{\pgfplotstablerowindex}]{\pgfplotstablerowindex}{\TransposedFilteredData}
    }
    {}
  }

  \begin{tikzpicture}
    \begin{axis} % [legend pos=outer north east]
      \addplot table [x=a, y=v] {\data};
      \addplot table [x=a, y={create col/linear regression={y=v}}] {\FilteredData}; % compute a linear regression from the input table
    \end{axis}
  \end{tikzpicture}
\end{document}

Как вы можете видеть из кода, я не очень опытен в программировании Latex. В любом случае, это работает. Всего наилучшего, Juhui

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